Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/
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;
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()
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()
/external/tensorflow/tensorflow/lite/kernels/
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 …]
Dkernel_util.cc585 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
/external/tensorflow/tensorflow/lite/tools/serialization/
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()
Dwriter_lib_test.cc392 TfLiteIntArrayView shape_view(tensor0->dims_signature); in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
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()
/external/tensorflow/tensorflow/lite/delegates/
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()
/external/tensorflow/tensorflow/lite/delegates/flex/
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()
/external/tensorflow/tensorflow/lite/delegates/coreml/
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);
/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/nnapi/
Dnnapi_compatibility_lib.cc44 for (int node_index : TfLiteIntArrayView(execution_plan)) { in CheckCompatibility()
/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/common/
Ddelegate_compatibility_checker_base.cc77 for (int node_index : TfLiteIntArrayView(execution_plan)) { in checkModelCompatibilityOnline()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
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 …]
Dnnapi_delegate_test.cc5506 for (auto d : TfLiteIntArrayView(tensor->dims_signature)) { in GetNNAPIDimensions()
/external/armnn/delegate/classic/src/
Darmnn_delegate.cpp484 for (auto inputIdx : tflite::TfLiteIntArrayView(tfLiteNode->inputs)) in Invoke()
502 for (auto outputIdx : tflite::TfLiteIntArrayView(tfLiteNode->outputs)) in Invoke()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_helper.cc288 return absl::StrJoin(TfLiteIntArrayView(dimensions), "x"); in GetDimensionString()
/external/tensorflow/tensorflow/lite/delegates/gpu/
Dmetal_delegate.mm296 for (int tensor_index : TfLiteIntArrayView(delegate_params->input_tensors)) {
321 for (int tensor_index : TfLiteIntArrayView(delegate_params->output_tensors)) {
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc128 return HasDynamicTensorImpl(context, TfLiteIntArrayView{int_array}, in HasDynamicTensor()