Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayView (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/
Dgraph_info.cc27 class TfLiteIntArrayView { class
31 explicit TfLiteIntArrayView(const TfLiteIntArray* int_array) in TfLiteIntArrayView() function in tflite::__anon3d89db6c0111::TfLiteIntArrayView
38 TfLiteIntArrayView(const TfLiteIntArrayView&) = default;
39 TfLiteIntArrayView& operator=(const TfLiteIntArrayView& rhs) = default;
61 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
78 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition()
145 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
165 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode()
170 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
Dcontext_util.h27 class TfLiteIntArrayView {
31 explicit TfLiteIntArrayView(const TfLiteIntArray* int_array) in TfLiteIntArrayView() function
34 TfLiteIntArrayView(const TfLiteIntArrayView&) = default;
35 TfLiteIntArrayView& operator=(const TfLiteIntArrayView& rhs) = default;
/external/tensorflow/tensorflow/lite/kernels/
Dwhile.cc134 TfLiteIntArrayView(node->inputs), in Prepare()
152 TfLiteIntArrayView(node->inputs), in Prepare()
237 CopyTensorsData(context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval()
292 TfLiteIntArrayView(node->outputs))); in Eval()
298 this_subgraph, TfLiteIntArrayView(node->outputs))); in Eval()
/external/tensorflow/tensorflow/lite/experimental/writer/
Dwriter_lib.cc105 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.inputs)); in ExportOperators()
107 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.outputs)); in ExportOperators()
132 TfLiteIntArrayView(node_and_registration->first.temporaries)) in ExportTensors()
182 TfLiteIntArrayView shape_view(tensor->dims); in ExportTensors()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc67 for (int index : TfLiteIntArrayView(indexes)) { in OpInputs()
115 for (int index : TfLiteIntArrayView(indexes)) { in OpOutputs()
385 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init()
391 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init()
399 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
Ddelegate.cc54 for (int node_index : TfLiteIntArrayView(plan)) { in Prepare()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc987 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
1065 for (auto absolute_input_index : TfLiteIntArrayView(node->inputs)) { in Invoke()
1090 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
1133 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
1233 for (auto input_index : TfLiteIntArrayView(node->inputs)) { in AddOpsAndTensors()
1252 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in AddOpsAndTensors()
1279 for (int i : TfLiteIntArrayView(input_tensors)) { in BuildGraph()
1290 for (int i : TfLiteIntArrayView(output_tensors)) { in BuildGraph()
1397 for (int node_index : TfLiteIntArrayView(plan)) { in NnApiDelegate()
/external/tensorflow/tensorflow/lite/g3doc/performance/
Ddelegates.md133 for (int node_index : TfLiteIntArrayView(plan)) {
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc72 return HasDynamicTensorImpl(context, TfLiteIntArrayView{int_array}); in HasDynamicTensor()