Searched refs:TfLiteIntArrayView (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/lite/ |
| D | context_util.h | 29 class TfLiteIntArrayView {
33 explicit TfLiteIntArrayView(const TfLiteIntArray* int_array) in TfLiteIntArrayView() function
36 TfLiteIntArrayView(const TfLiteIntArrayView&) = default;
37 TfLiteIntArrayView& operator=(const TfLiteIntArrayView& rhs) = default;
|
| D | graph_info.cc | 46 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
64 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition()
131 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
153 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode()
158 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | while.cc | 156 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare()
173 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare()
262 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval()
268 CopyTensorsData(context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval()
323 this_subgraph, TfLiteIntArrayView(node->outputs), false)); in Eval()
329 this_subgraph, TfLiteIntArrayView(node->outputs))); in Eval()
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 144 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.inputs)); in ExportOperators()
146 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.outputs)); in ExportOperators()
171 TfLiteIntArrayView(node_and_registration->first.temporaries)) in ExportTensors()
235 TfLiteIntArrayView shape_view(tensor->dims); in ExportTensors()
339 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput()
363 TfLiteIntArrayView outputs(node.outputs); in CheckInputOutput()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/ |
| D | hexagon_delegate_kernel.cc | 80 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
124 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Eval()
211 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Prepare()
214 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Prepare()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 58 return absl::StrCat("[", absl::StrJoin(tflite::TfLiteIntArrayView(dims), ","), in GetDimsDebugString()
78 for (int index : TfLiteIntArrayView(indexes)) { in OpInputs()
126 for (int index : TfLiteIntArrayView(indexes)) { in OpOutputs()
387 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init()
393 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init()
401 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
|
| /external/tensorflow/tensorflow/lite/delegates/coreml/ |
| D | coreml_delegate_kernel.mm | 124 for (int node_index : TfLiteIntArrayView(delegate_params->nodes_to_replace)) {
175 for (int tensor_index : TfLiteIntArrayView(node->inputs)) {
197 for (int tensor_index : TfLiteIntArrayView(node->outputs)) {
209 TfLiteIntArrayView node_inputs(node->inputs);
|
| /external/tensorflow/tensorflow/lite/delegates/ |
| D | utils.cc | 126 for (int node_id : TfLiteIntArrayView(execution_plan)) { in PrepareSupportedNodes()
286 auto inputs_view = TfLiteIntArrayView(inputs); in RemapFp16InputTensors()
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 386 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
517 for (auto i : TfLiteIntArrayView(int_array)) { in GetHash()
523 bool HasZeroes(TfLiteIntArrayView array) { in HasZeroes()
2163 const TfLiteIntArrayView input_shape( in Validate()
3619 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
3674 for (int i : TfLiteIntArrayView(params->input_tensors)) { in Init()
3862 for (int i : TfLiteIntArrayView(node->inputs)) { in Invoke()
3893 for (int i : TfLiteIntArrayView(node->outputs)) { in Invoke()
3914 for (auto absolute_input_index : TfLiteIntArrayView(node->inputs)) { in Invoke()
4015 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.cc | 286 return absl::StrJoin(TfLiteIntArrayView(dimensions), "x"); in GetDimensionString()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/ |
| D | metal_delegate.mm | 296 for (int tensor_index : TfLiteIntArrayView(delegate_params->input_tensors)) {
321 for (int tensor_index : TfLiteIntArrayView(delegate_params->output_tensors)) {
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 114 return HasDynamicTensorImpl(context, TfLiteIntArrayView{int_array}); in HasDynamicTensor()
|