| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 38 TensorRef<BHWC> tensor_ref; in TEST() local
40 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
43 EXPECT_EQ(tensor_ref.type, DataType::FLOAT32); in TEST()
44 EXPECT_EQ(tensor_ref.shape, BHWC(4, 1, 1, 1)); in TEST()
54 TensorRef<BHWC> tensor_ref; in TEST() local
56 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
59 EXPECT_EQ(tensor_ref.type, DataType::INT32); in TEST()
60 EXPECT_EQ(tensor_ref.shape, BHWC(4, 1, 1, 5)); in TEST()
71 TensorRef<BHWC> tensor_ref; in TEST() local
73 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
[all …]
|
| D | model_builder.h | 47 TensorRef<BHWC>* tensor_ref);
|
| D | model_builder.cc | 2616 TensorRef<BHWC>* tensor_ref) { in ConvertTfLiteTensorToTensorRef() argument
2617 tensor_ref->type = ToDataType(tflite_tensor.type); in ConvertTfLiteTensorToTensorRef()
2618 return ExtractTensorShape(tflite_tensor, &tensor_ref->shape); in ConvertTfLiteTensorToTensorRef()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/ |
| D | relu_test.cc | 37 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
38 tensor_ref.type = DataType::FLOAT32; in GetTensorRef()
39 tensor_ref.ref = ref; in GetTensorRef()
40 tensor_ref.shape = BHWC(1, 2, 2, 1); in GetTensorRef()
41 return tensor_ref; in GetTensorRef()
|
| D | elementwise_test.cc | 32 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
33 tensor_ref.type = DataType::FLOAT32; in GetTensorRef()
34 tensor_ref.ref = ref; in GetTensorRef()
35 tensor_ref.shape = shape; in GetTensorRef()
36 return tensor_ref; in GetTensorRef()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | unique_tensor_references.cc | 40 const TensorReference tensor_ref(tensor); in Add() local
41 if (!referenced_tensors_set_->insert(tensor_ref).second) { in Add()
43 tensor_ref.Unref(); in Add()
|
| D | op_kernel.cc | 879 TensorReference tensor_ref(*t); in allocate_persistent() local
880 tensor_ref.FillDescription(&alloc_desc); in allocate_persistent()
881 tensor_ref.Unref(); in allocate_persistent()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | relu_test.mm | 48 TensorRef<BHWC> tensor_ref;
49 tensor_ref.type = DataType::FLOAT32;
50 tensor_ref.ref = ref;
51 tensor_ref.shape = BHWC(1, 2, 2, 1);
52 return tensor_ref;
|
| D | elementwise_test.mm | 47 TensorRef<BHWC> tensor_ref;
48 tensor_ref.type = DataType::FLOAT32;
49 tensor_ref.ref = ref;
50 tensor_ref.shape = shape;
51 return tensor_ref;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_device_array.h | 81 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()
91 stream, [tensor_ref]() { tensor_ref.Unref(); }); in Finalize()
|
| D | rocm_solvers.cc | 160 for (auto tensor_ref : scratch_tensor_refs_) { in ~ROCmSolver() local
161 tensor_ref.Unref(); in ~ROCmSolver()
|
| D | cuda_solvers.cc | 173 for (auto tensor_ref : scratch_tensor_refs_) { in ~CudaSolver() local
174 tensor_ref.Unref(); in ~CudaSolver()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node.cc | 84 for (const auto& tensor_ref : protected_tensors_) { in ~ExecuteNodeArgs() local
85 tensor_ref.Unref(); in ~ExecuteNodeArgs()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | object_manager.cc | 35 Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref, in CreatePHWC4BufferFromTensorRef() argument
38 GetElementsSizeForPHWC4(tensor_ref.shape), gl_buffer); in CreatePHWC4BufferFromTensorRef()
|
| D | object_manager.h | 75 Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref,
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 164 TensorReference tensor_ref(tensor); in SetProtoFromGPU() local
167 total_bytes, alloc, tensor_ref]() { in SetProtoFromGPU()
171 tensor_ref.Unref(); in SetProtoFromGPU()
|