Searched refs:tensor_allocator (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/ |
| D | main.cc | 53 tflite::SimpleTensorAllocator tensor_allocator(tensor_arena, in main() local
57 tflite::MicroInterpreter interpreter(model, resolver, &tensor_allocator, in main()
|
| D | micro_speech_test.cc | 50 tflite::SimpleTensorAllocator tensor_allocator(tensor_arena, in TF_LITE_MICRO_TEST() local
54 tflite::MicroInterpreter interpreter(model, resolver, &tensor_allocator, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/examples/micro_speech/apollo3/ |
| D | pushbutton_test.cc | 78 tflite::SimpleTensorAllocator tensor_allocator(tensor_arena, in TF_LITE_MICRO_TEST() local
82 tflite::MicroInterpreter interpreter(model, resolver, &tensor_allocator, in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | i_remote_fused_graph_executor.h | 62 TensorAllocatorFunc tensor_allocator) = 0;
|
| D | remote_fused_graph_execute_op_test.cc | 207 TensorAllocatorFunc tensor_allocator) final { in ReadOutputNode() argument
211 Tensor* output_tensor = tensor_allocator(output_shape); in ReadOutputNode()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/ |
| D | micro_interpreter.h | 37 SimpleTensorAllocator* tensor_allocator,
|
| D | micro_interpreter.cc | 63 SimpleTensorAllocator* tensor_allocator, in MicroInterpreter() argument
67 tensor_allocator_(tensor_allocator), in MicroInterpreter()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | hexagon_control_wrapper.h | 50 TensorAllocatorFunc tensor_allocator) final;
|
| D | hexagon_control_wrapper.cc | 355 const string& node_name, TensorAllocatorFunc tensor_allocator) { in ReadOutputNode() argument
372 Tensor* output_tensor = tensor_allocator(output_shape); in ReadOutputNode()
|