Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dcontext_util.h29 class TfLiteIntArrayView {
33 explicit TfLiteIntArrayView(const TfLiteIntArray* int_array) in TfLiteIntArrayView() function
36 TfLiteIntArrayView(const TfLiteIntArrayView&) = default;
37 TfLiteIntArrayView& operator=(const TfLiteIntArrayView& rhs) = default;
H A Doptional_debug_tools.cc462 for (int nid : TfLiteIntArrayView(params->nodes_to_replace)) { in PrintInterpreterState()
504 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
505 : TfLiteIntArrayView(node.inputs)); in PrintInterpreterState()
511 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
512 : TfLiteIntArrayView(node.outputs)); in PrintInterpreterState()
519 ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
520 : TfLiteIntArrayView(node.intermediates)); in PrintInterpreterState()
527 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
528 : TfLiteIntArrayView(node.temporaries)); in PrintInterpreterState()
H A Dgraph_info.cc54 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
79 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition()
141 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
168 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode()
173 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dwhile.cc254 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare_impl()
271 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare_impl()
420 TfLiteIntArrayView(node->inputs), in Eval_dynamic()
426 TfLiteIntArrayView(node->inputs), in Eval_dynamic()
428 TfLiteIntArrayView(node->outputs))); in Eval_dynamic()
445 TfLiteIntArrayView(node->outputs), in Eval_dynamic()
450 TfLiteIntArrayView(node->outputs), in Eval_dynamic()
470 this_subgraph, TfLiteIntArrayView(node->outputs))); in Eval_dynamic()
527 CopyTensorsData(context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval_static()
552 TfLiteIntArrayView(node->inputs), in Eval_static()
[all …]
H A Dkernel_util.cc585 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/
H A Dwriter_lib.cc149 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.inputs)); in ExportOperators()
151 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.outputs)); in ExportOperators()
176 TfLiteIntArrayView(node_and_registration->first.temporaries)) in ExportTensors()
244 TfLiteIntArrayView shape_view(tensor->dims); in ExportTensors()
256 TfLiteIntArrayView shape_signature_view(tensor->dims_signature); in ExportTensors()
366 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput()
390 TfLiteIntArrayView outputs(node.outputs); in CheckInputOutput()
H A Dwriter_lib_test.cc392 TfLiteIntArrayView shape_view(tensor0->dims_signature); in TEST_P()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/
H A Dhexagon_delegate_kernel.cc82 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
126 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Eval()
215 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Prepare()
218 for (auto tensor_index : TfLiteIntArrayView(node->outputs)) { in Prepare()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc83 for (int index : TfLiteIntArrayView(indexes)) { in OpInputs()
131 for (int index : TfLiteIntArrayView(indexes)) { in OpOutputs()
455 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init()
461 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init()
480 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
503 for (auto tensor_index : TfLiteIntArrayView(node->inputs)) { in Init()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/
H A Dcoreml_delegate_kernel.mm124 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)) {
210 TfLiteIntArrayView node_inputs(node->inputs);
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/
H A Dutils.cc135 for (int node_id : TfLiteIntArrayView(original_execution_plan_)) { in PrepareSupportedNodes()
176 for (int node_id : TfLiteIntArrayView(original_execution_plan_)) { in GetNodesOfFirstNLargestPartitionsImpl()
261 auto inputs_view = TfLiteIntArrayView(inputs); in RemapFp16InputTensors()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/nnapi/
H A Dnnapi_compatibility_lib.cc44 for (int node_index : TfLiteIntArrayView(execution_plan)) { in CheckCompatibility()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc593 for (auto i : TfLiteIntArrayView(int_array)) { in GetHash()
599 bool HasZeroes(TfLiteIntArrayView array) { in HasZeroes()
2170 for (int i : TfLiteIntArrayView(tensor_indices)) { in AppendDynamicDimensions()
2815 const TfLiteIntArrayView input_shape( in Validate()
3533 const TfLiteIntArrayView input_shape( in Validate()
4467 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
4841 for (int i : TfLiteIntArrayView(node->inputs)) { in Invoke()
4876 for (int i : TfLiteIntArrayView(node->outputs)) { in Invoke()
4918 for (auto absolute_input_index : TfLiteIntArrayView(node->inputs)) { in Invoke()
5044 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
[all …]
H A Dnnapi_delegate_test.cc5506 for (auto d : TfLiteIntArrayView(tensor->dims_signature)) { in GetNNAPIDimensions()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/common/
H A Ddelegate_compatibility_checker_base.cc77 for (int node_index : TfLiteIntArrayView(execution_plan)) { in checkModelCompatibilityOnline()
/aosp_15_r20/external/armnn/delegate/classic/src/
H A Darmnn_delegate.cpp484 for (auto inputIdx : tflite::TfLiteIntArrayView(tfLiteNode->inputs)) in Invoke()
502 for (auto outputIdx : tflite::TfLiteIntArrayView(tfLiteNode->outputs)) in Invoke()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_helper.cc288 return absl::StrJoin(TfLiteIntArrayView(dimensions), "x"); in GetDimensionString()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/
H A Dmetal_delegate.mm296 for (int tensor_index : TfLiteIntArrayView(delegate_params->input_tensors)) {
321 for (int tensor_index : TfLiteIntArrayView(delegate_params->output_tensors)) {
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc128 return HasDynamicTensorImpl(context, TfLiteIntArrayView{int_array}, in HasDynamicTensor()
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow/lite/kernels/libtflite_kernel_utils/android_arm64_armv8-2a_cortex-a55_static_apex30_sabi/obj/external/tensorflow/tensorflow/lite/kernels/
Dkernel_util.sdump5211 "function_name" : "tflite::TfLiteIntArrayView::TfLiteIntArrayView",
5227 "function_name" : "tflite::TfLiteIntArrayView::TfLiteIntArrayView",
5243 "function_name" : "tflite::TfLiteIntArrayView::TfLiteIntArrayView",
5259 "function_name" : "tflite::TfLiteIntArrayView::TfLiteIntArrayView",
5275 "function_name" : "tflite::TfLiteIntArrayView::operator=",
6144 "function_name" : "tflite::TfLiteIntArrayView::end",
6157 "function_name" : "tflite::TfLiteIntArrayView::size",
6170 "function_name" : "tflite::TfLiteIntArrayView::begin",
6183 "function_name" : "tflite::TfLiteIntArrayView::operator[]",
52800 "name" : "const tflite::TfLiteIntArrayView &",
[all …]