Searched refs:new_tensor (Results 1 – 8 of 8) sorted by relevance
/external/eigen/unsupported/test/ |
D | cxx11_tensor_assign.cpp | 238 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/ |
D | convert_to_constants.py | 74 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)
|
D | ops.py | 847 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/ |
D | 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/ |
D | __init__.py | 164 new_tensor = (tf.div(summed_tensor, mpi_size) 166 return new_tensor
|
/external/tensorflow/tensorflow/core/framework/ |
D | op_kernel.cc | 455 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/ |
D | indexed_array_analysis.h | 331 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/ |
D | trt_convert.py | 290 new_tensor = graph.get_tensor_by_name(tensor.name) 291 new_tensor.set_shape(tensor.shape) 292 new_tensors.append(new_tensor)
|