Home
last modified time | relevance | path

Searched refs:TfLiteIntArrayView (Results 1 – 14 of 14) 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.cc46 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl()
80 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in Partition()
147 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
175 for (int output_tensor_index : TfLiteIntArrayView(node.outputs)) { in UpdateNode()
180 for (int input_tensor_index : TfLiteIntArrayView(node.inputs)) { in UpdateNode()
Doptional_debug_tools.cc414 for (int nid : TfLiteIntArrayView(params->nodes_to_replace)) { in PrintInterpreterState()
456 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
457 : TfLiteIntArrayView(node.inputs)); in PrintInterpreterState()
463 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
464 : TfLiteIntArrayView(node.outputs)); in PrintInterpreterState()
471 ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
472 : TfLiteIntArrayView(node.intermediates)); in PrintInterpreterState()
479 subgraph, is_node_delegated ? TfLiteIntArrayView(&empty_int_array) in PrintInterpreterState()
480 : TfLiteIntArrayView(node.temporaries)); in PrintInterpreterState()
/external/tensorflow/tensorflow/lite/kernels/
Dwhile.cc158 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare()
175 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Prepare()
264 context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval()
270 CopyTensorsData(context, this_subgraph, TfLiteIntArrayView(node->inputs), in Eval()
325 this_subgraph, TfLiteIntArrayView(node->outputs), false)); in Eval()
331 this_subgraph, TfLiteIntArrayView(node->outputs))); in Eval()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc80 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/tools/serialization/
Dwriter_lib.cc144 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.inputs)); in ExportOperators()
146 RemapTensorIndicesToWritten(TfLiteIntArrayView(node.outputs)); in ExportOperators()
171 TfLiteIntArrayView(node_and_registration->first.temporaries)) in ExportTensors()
237 TfLiteIntArrayView shape_view(tensor->dims); in ExportTensors()
348 for (int tensor_index : TfLiteIntArrayView(node.inputs)) { in CheckInputOutput()
372 TfLiteIntArrayView outputs(node.outputs); in CheckInputOutput()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dkernel.cc58 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()
393 for (auto tensor_index : TfLiteIntArrayView(params->output_tensors)) { in Init()
399 for (auto tensor_index : TfLiteIntArrayView(params->input_tensors)) { in Init()
407 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
/external/tensorflow/tensorflow/lite/delegates/
Dutils.cc134 for (int node_id : TfLiteIntArrayView(original_execution_plan_)) { in PrepareSupportedNodes()
175 for (int node_id : TfLiteIntArrayView(original_execution_plan_)) { in GetNodesOfFirstNLargestPartitionsImpl()
260 auto inputs_view = TfLiteIntArrayView(inputs); in RemapFp16InputTensors()
/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)) {
209 TfLiteIntArrayView node_inputs(node->inputs);
/external/armnn/delegate/src/
Darmnn_delegate.cpp372 for (auto inputIdx : tflite::TfLiteIntArrayView(tfLiteNode->inputs)) in Invoke()
388 for (auto outputIdx : tflite::TfLiteIntArrayView(tfLiteNode->outputs)) in Invoke()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc431 for (int i : TfLiteIntArrayView(tensor->dims_signature)) { in HasUnspecifiedDimension()
572 for (auto i : TfLiteIntArrayView(int_array)) { in GetHash()
578 bool HasZeroes(TfLiteIntArrayView array) { in HasZeroes()
2382 const TfLiteIntArrayView input_shape( in Validate()
3869 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init()
4184 for (int i : TfLiteIntArrayView(node->inputs)) { in Invoke()
4216 for (int i : TfLiteIntArrayView(node->outputs)) { in Invoke()
4237 for (auto absolute_input_index : TfLiteIntArrayView(node->inputs)) { in Invoke()
4361 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
4459 for (auto output_index : TfLiteIntArrayView(node->outputs)) { in Invoke()
[all …]
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_helper.cc286 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.cc120 return HasDynamicTensorImpl(context, TfLiteIntArrayView{int_array}); in HasDynamicTensor()