Home
last modified time | relevance | path

Searched refs:new_tensor (Results 1 – 8 of 8) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_assign.cpp238 Tensor<int, 1> new_tensor(5); in test_auto_resize() local
239 new_tensor.setRandom(); in test_auto_resize()
241 tensor1 = tensor2 = tensor3 = tensor4 = new_tensor; in test_auto_resize()
243 VERIFY_IS_EQUAL(tensor1.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
244 VERIFY_IS_EQUAL(tensor2.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
245 VERIFY_IS_EQUAL(tensor3.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
246 VERIFY_IS_EQUAL(tensor4.dimension(0), new_tensor.dimension(0)); in test_auto_resize()
247 for (int i = 0; i < new_tensor.dimension(0); ++i) { in test_auto_resize()
248 VERIFY_IS_EQUAL(tensor1(i), new_tensor(i)); in test_auto_resize()
249 VERIFY_IS_EQUAL(tensor2(i), new_tensor(i)); in test_auto_resize()
[all …]
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py74 new_tensor = graph.get_tensor_by_name(orig_tensor.name)
75 if new_tensor.shape.rank is None:
76 new_tensor.set_shape(orig_tensor.shape)
77 new_tensors.append(new_tensor)
Dops.py847 new_tensor = self._copy_to_device(context=ctx._handle, device=device_name)
850 return new_tensor
854 new_tensor = self._copy_nograd(ctx, device_name)
860 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
861 return new_tensor
/external/tensorflow/tensorflow/contrib/copy_graph/python/util/
Dcopy_elements.py171 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/
D__init__.py164 new_tensor = (tf.div(summed_tensor, mpi_size)
166 return new_tensor
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.cc455 std::unique_ptr<Tensor> new_tensor = forward_input( in forward_input_to_output_with_shape() local
458 if (new_tensor != nullptr) { in forward_input_to_output_with_shape()
460 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
569 std::unique_ptr<Tensor> new_tensor = in forward_input_or_allocate_temp() local
572 if (new_tensor != nullptr) { in forward_input_or_allocate_temp()
573 *out_temp = std::move(*new_tensor); in forward_input_or_allocate_temp()
708 Tensor new_tensor(a, type, shape, logged_attr); in allocate_tensor() local
710 if (!new_tensor.IsInitialized()) { in allocate_tensor()
718 params_->step_id, new_tensor); in allocate_tensor()
720 record_tensor_reference(new_tensor); in allocate_tensor()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.h331 T* new_tensor = new T(std::forward<Args>(args)...); in Construct() local
332 owned_tensors_.push_back(std::unique_ptr<T>(new_tensor)); in Construct()
333 return new_tensor; in Construct()
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py290 new_tensor = graph.get_tensor_by_name(tensor.name)
291 new_tensor.set_shape(tensor.shape)
292 new_tensors.append(new_tensor)