Searched refs:new_tensor (Results 1 – 5 of 5) 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/core/framework/ |
| D | op_kernel.cc | 487 std::unique_ptr<Tensor> new_tensor = forward_input( in forward_input_to_output_with_shape() local
490 if (new_tensor != nullptr) { in forward_input_to_output_with_shape()
492 outputs_[output_index] = TensorValue(new_tensor.release()); in forward_input_to_output_with_shape()
601 std::unique_ptr<Tensor> new_tensor = in forward_input_or_allocate_temp() local
604 if (new_tensor != nullptr) { in forward_input_or_allocate_temp()
605 *out_temp = std::move(*new_tensor); in forward_input_or_allocate_temp()
753 Tensor new_tensor(a, type, shape, in allocate_tensor() local
758 if (!new_tensor.IsInitialized()) { in allocate_tensor()
766 params_->step_id, new_tensor); in allocate_tensor()
768 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/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 866 TfLiteTensor* new_tensor = &context_->tensors[*tensor_index]; in AddNewInputConstantTensor() local
867 new_tensor->type = type; in AddNewInputConstantTensor()
868 new_tensor->allocation_type = kTfLiteDynamic; in AddNewInputConstantTensor()
869 new_tensor->params = quant_params; in AddNewInputConstantTensor()
876 context_, new_tensor, in AddNewInputConstantTensor()
880 memcpy(new_tensor->data.raw, in AddNewInputConstantTensor()
903 nn_model_, ann_tensor_index, new_tensor->data.raw, in AddNewInputConstantTensor()
904 new_tensor->bytes), in AddNewInputConstantTensor()
1155 TfLiteTensor* new_tensor = &context_->tensors[new_tensor_index]; in AddTensor() local
1156 new_tensor->type = kTfLiteUInt8; in AddTensor()
[all …]
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 1046 new_tensor = self._copy_to_device(device_name)
1049 return new_tensor
1053 new_tensor = self._copy_nograd(ctx, device_name)
1064 tape.record_operation("_copy", [new_tensor], [self], grad_fun)
1065 return new_tensor
|