Searched refs:SetBufferHandle (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_errno_test.cc | 41 void SetBufferHandle(int index, TfLiteBufferHandle handle) { in SetBufferHandle() function in tflite::__anon406f5a1c0111::SingleOpModelWithNNAPI
42 interpreter_->SetBufferHandle(index, handle, stateful_delegate_.get()); in SetBufferHandle()
|
| D | nnapi_delegate_signed_quantization_test.cc | 51 void SetBufferHandle(int index, TfLiteBufferHandle handle) { in SetBufferHandle() function in tflite::__anonb9fc30f60111::SingleOpModelWithNNAPI
52 interpreter_->SetBufferHandle(index, handle, stateful_delegate_.get()); in SetBufferHandle()
|
| D | nnapi_delegate_test.cc | 88 void SetBufferHandle(int index, TfLiteBufferHandle handle) { in SetBufferHandle() function in tflite::__anon0c68fcb70111::SingleOpModelWithNNAPI
89 interpreter_->SetBufferHandle(index, handle, stateful_delegate_.get()); in SetBufferHandle()
466 m.SetBufferHandle(m.input1(), input1_handle); in TEST()
478 m.SetBufferHandle(m.input1(), input1_handle); in TEST()
485 m.SetBufferHandle(m.input1(), kTfLiteNullBufferHandle); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | metal_spatial_tensor.h | 84 absl::Status SetBufferHandle(id<MTLBuffer> buffer);
|
| D | metal_spatial_tensor.cc | 524 absl::Status MetalSpatialTensor::SetBufferHandle(id<MTLBuffer> buffer) { in SetBufferHandle() function in tflite::gpu::metal::MetalSpatialTensor
|
| D | inference_context.cc | 659 auto status = preallocated_tensors_[it.first].SetBufferHandle(it.second); in UpdatePreallocatedTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | delegate_test.cc | 283 interpreter_->SetBufferHandle(kOutputTensorIndex, handle, delegate); in TEST_F()
303 interpreter_->SetBufferHandle(kOutputTensorIndex, handle, delegate); in TEST_F()
325 TfLiteStatus status = interpreter_->SetBufferHandle( in TEST_F()
694 interpreter_->SetBufferHandle(kOutputTensorIndex, handle, delegate); in TEST_F()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter.cc | 415 TfLiteStatus Interpreter::SetBufferHandle(int tensor_index, in SetBufferHandle() function in tflite::Interpreter
|
| D | interpreter.h | 536 TfLiteStatus SetBufferHandle(int tensor_index,
|