| /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 | 33 TensorRef<BHWC> tensor_ref; in GetTensorRef() local
34 tensor_ref.type = DataType::FLOAT32; in GetTensorRef()
35 tensor_ref.ref = ref; in GetTensorRef()
36 tensor_ref.shape = shape; in GetTensorRef()
37 return tensor_ref; in GetTensorRef()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_test.cc | 44 TensorRef<BHWC> tensor_ref; in TEST() local
46 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
49 EXPECT_EQ(tensor_ref.type, DataType::FLOAT32); in TEST()
50 EXPECT_EQ(tensor_ref.shape, BHWC(4, 1, 1, 1)); in TEST()
60 TensorRef<BHWC> tensor_ref; in TEST() local
62 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
65 EXPECT_EQ(tensor_ref.type, DataType::INT32); in TEST()
66 EXPECT_EQ(tensor_ref.shape, BHWC(4, 1, 1, 5)); in TEST()
77 TensorRef<BHWC> tensor_ref; in TEST() local
79 ConvertTfLiteTensorToTensorRef(tflite_tensor, &tensor_ref); in TEST()
[all …]
|
| D | model_builder.h | 68 TensorRef<BHWC>* tensor_ref);
|
| D | model_builder_helper.cc | 157 TensorRef<BHWC>* tensor_ref) { in ConvertTfLiteTensorToTensorRef() argument
158 tensor_ref->type = ToDataType(tflite_tensor.type); in ConvertTfLiteTensorToTensorRef()
159 return ExtractTensorShape(tflite_tensor, &tensor_ref->shape); in ConvertTfLiteTensorToTensorRef()
|
| D | model_builder_helper.h | 52 TensorRef<BHWC>* tensor_ref);
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gpu_device_array.h | 82 TensorReference tensor_ref(out_of_line_values_on_host_); in Finalize()
92 stream, [tensor_ref]() { tensor_ref.Unref(); }); in Finalize()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/gl/ |
| D | object_manager.cc | 35 absl::Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref, in CreatePHWC4BufferFromTensorRef() argument
38 GetElementsSizeForPHWC4(tensor_ref.shape), gl_buffer); in CreatePHWC4BufferFromTensorRef()
|
| D | object_manager.h | 75 absl::Status CreatePHWC4BufferFromTensorRef(const TensorRef<BHWC>& tensor_ref,
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | dlpack.cc | 270 TensorReference tensor_ref(*tensor); // This will call buf_->Ref() in TFE_HandleToDLPack() local
271 auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref); in TFE_HandleToDLPack()
272 tf_dlm_tensor_ctx->reference = tensor_ref; in TFE_HandleToDLPack()
|
| /external/tensorflow/tensorflow/core/util/ |
| 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 | 194 for (const auto& tensor_ref : scratch_tensor_refs_) { in ~CudaSolver() local
195 tensor_ref.Unref(); in ~CudaSolver()
|
| /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()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2020-024.md | 18 auto* tf_dlm_tensor_ctx = new TfDlManagedTensorCtx(tensor_ref);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | op_kernel.cc | 842 TensorReference tensor_ref(*t); in allocate_persistent() local
843 tensor_ref.FillDescription(&alloc_desc); in allocate_persistent()
844 tensor_ref.Unref(); in allocate_persistent()
|