HomeSort by relevance Sort by last modified time
    Searched refs:new_tensor (Results 1 - 5 of 5) sorted by null

  /external/eigen/unsupported/test/
cxx11_tensor_assign.cpp 238 Tensor<int, 1> new_tensor(5);
239 new_tensor.setRandom();
241 tensor1 = tensor2 = tensor3 = tensor4 = new_tensor;
243 VERIFY_IS_EQUAL(tensor1.dimension(0), new_tensor.dimension(0));
244 VERIFY_IS_EQUAL(tensor2.dimension(0), new_tensor.dimension(0));
245 VERIFY_IS_EQUAL(tensor3.dimension(0), new_tensor.dimension(0));
246 VERIFY_IS_EQUAL(tensor4.dimension(0), new_tensor.dimension(0));
247 for (int i = 0; i < new_tensor.dimension(0); ++i) {
248 VERIFY_IS_EQUAL(tensor1(i), new_tensor(i));
249 VERIFY_IS_EQUAL(tensor2(i), new_tensor(i))
    [all...]
  /external/tensorflow/tensorflow/contrib/copy_graph/python/util/
copy_elements.py 171 new_tensor = new_op.outputs[output_index]
174 to_graph.add_to_collection(collection, new_tensor)
176 return new_tensor
  /external/tensorflow/tensorflow/contrib/mpi_collectives/
__init__.py 164 new_tensor = (tf.div(summed_tensor, mpi_size)
166 return new_tensor
  /external/tensorflow/tensorflow/core/framework/
op_kernel.cc 417 std::unique_ptr<Tensor> new_tensor = forward_input( local
420 if (new_tensor != nullptr) {
422 outputs_[output_index] = TensorValue(new_tensor.release());
502 std::unique_ptr<Tensor> new_tensor = local
504 if (new_tensor != nullptr) {
505 *out_temp = std::move(*new_tensor);
632 Tensor new_tensor(a, type, shape, logged_attr);
634 if (!new_tensor.IsInitialized()) {
642 params_->step_id, new_tensor);
644 record_tensor_reference(new_tensor);
    [all...]
  /external/tensorflow/tensorflow/python/framework/
ops.py 770 new_tensor = self._copy_to_device(context=ctx._handle, device=device_name)
779 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
780 return new_tensor
    [all...]

Completed in 483 milliseconds