Home
last modified time | relevance | path

Searched defs:node_index (Results 1 – 25 of 82) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/xnnpack/
H A Dxnnpack_delegate.cc296 const int node_index = params->nodes_to_replace->data[i]; in Create() local
664 const int node_index = params->nodes_to_replace->data[i]; in Create() local
829 for (size_t node_index = 0; node_index < num_operators; ++node_index) { in AddEventsToProfiler() local
841 int node_index) { in CalculatePadding()
861 int dilation_width, int stride_height, int stride_width, int node_index, in CalculateTransposeConvPaddings()
937 TfLiteContext* context, int node_index, TfLiteFusedActivation activation, in ConvertActivationToOutputRange()
981 int node_index) { in CheckConvolutionParams()
1011 int output_channels, int node_index) { in CheckDepthwiseConvolutionParams()
1056 int node_index) { in CheckMediaPipeTransposedConvolutionParams()
1073 int node_index) { in CheckMediaPipePoolParams()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dsubgraph_test_util.cc115 int node_index; in BuildAddSubgraph() local
146 int node_index; in BuildMulSubgraph() local
176 int node_index; in BuildPadSubgraph() local
211 int node_index; in BuildIfSubgraph() local
244 int node_index; in BuildLessEqualCondSubgraph() local
279 int node_index; in BuildAccumulateLoopBodySubgraph() local
331 int node_index; in BuildPadLoopBodySubgraph() local
379 int node_index; in BuildWhileSubgraph() local
403 int node_index; in BuildAssignRandomValueToVariableSubgraph() local
434 int node_index; in BuildCallOnceAndReadVariableSubgraph() local
[all …]
H A Dvariable_ops_test.cc66 int node_index; in ConstructInvalidGraph() local
101 int node_index; in ConstructGraph() local
132 int node_index; in ConstructGraphWithKnownShape() local
H A Dhashtable_ops_test.cc120 int node_index; in BuildDefaultGraph() local
147 int node_index; in BuildNoImportGraph() local
169 int node_index; in BuildImportTwiceGraph() local
201 int node_index; in BuildTwoHashtablesGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/utils/
H A Dgraph_view.h48 FaninView(GraphView* graph_view, int node_index, int port_index) in FaninView()
65 FanoutView(GraphView* graph_view, int node_index, int port_index) in FanoutView()
82 explicit NodeView(GraphView* graph_view, int node_index) in NodeView()
143 MutableFaninView(MutableGraphView* graph_view, int node_index, int port_index) in MutableFaninView()
146 explicit MutableFaninView(MutableGraphView* graph_view, int node_index, in MutableFaninView()
178 MutableFanoutView(MutableGraphView* graph_view, int node_index, in MutableFanoutView()
182 explicit MutableFanoutView(MutableGraphView* graph_view, int node_index, in MutableFanoutView()
211 explicit MutableNodeView(MutableGraphView* graph_view, int node_index) in MutableNodeView()
H A Dgraph_view_internal.h49 NodeIndexAndPortIndex(GraphViewT* graph_view, int node_index, int port_index) in NodeIndexAndPortIndex()
73 int node_index() const { return node_index_; } in node_index() function
130 explicit NodeViewInternal(GraphViewT* graph_view, int node_index) in NodeViewInternal()
156 int node_index() const { return node_index_; } in node_index() function
293 const NodeViewT* GetNode(int node_index) const { in GetNode()
301 NodeViewT* GetNode(int node_index) { in GetNode()
370 int node_index; member
H A Dgraph_view.cc111 const int node_index = node_index_by_name_.size(); in AddUniqueNodeInternal() local
124 const int node_index = node_view->node_index_; in CheckAndAddFaninsInternal() local
493 const int node_index = node_index_by_name_.size(); in AddUniqueNodeInternal() local
621 for (int node_index : mutation_.removed_nodes_) { in GetNodeNamesAndPartitionUpdatedNodes() local
951 int node_index; in AddNewNodes() local
1325 for (int node_index : mutation_.removed_nodes_) { in RemoveNodesInternal() local
1413 const int node_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc80 int node_index, const char* message) { in ReportOpError()
211 int node_index = subgraph_->execution_plan()[index]; in node() local
214 size_t node_index(size_t index) const override { in node_index() function in tflite::InterpreterInfo
271 for (int node_index = 0; node_index < nodes_and_registration_.size(); in ~Subgraph() local
287 void Subgraph::CleanupNode(int node_index) { in CleanupNode()
444 int node_index; in ReplaceNodeSubsetsWithDelegateKernels() local
796 const TfLiteRegistration* registration, int* node_index) { in AddNodeWithParameters()
1087 int node_index = execution_plan[execution_plan_index]; in PrepareOpsStartingAt() local
1259 int node_index = execution_plan_[execution_plan_index]; in Invoke() local
1432 int node_index, TfLiteNode** node, TfLiteRegistration** registration) { in GetNodeAndRegistration()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dgraph_info.cc54 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl() local
76 for (int node_index = 0; node_index < info_->num_execution_nodes(); in Partition() local
131 bool UpdateNode(int node_index) { in UpdateNode()
212 for (int node_index = 0; node_index < info_->num_execution_nodes(); in BuildNodeSubset() local
263 for (int last_op_with_side_effect = -1, node_index = 0; in PartitionGraphIntoIndependentNodeSubsets() local
H A Dtensorflow_profiler_logger_shim.cc30 int node_index) {} in OnTfLiteOpPrepare()
47 const char* op_name, int subgraph_index, int node_index) { in OnTfLiteOpInvoke()
H A Dtensorflow_profiler_logger.cc113 int node_index) { in OnTfLiteOpPrepare()
139 int node_index) { in OnTfLiteOpInvoke()
H A Dinterpreter.h263 int node_index) const { in node_and_registration()
270 int subgraph_index, int node_index) const { in node_and_registration()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc658 bool FindContractionWithBias(const RemapperContext& ctx, int node_index, in FindContractionWithBias()
702 const RemapperContext& ctx, Cluster* cluster, int node_index, in FindContractionWithBiasAndActivation()
753 bool FindConvWithSqueezeAndBias(const RemapperContext& ctx, int node_index, in FindConvWithSqueezeAndBias()
811 bool FindConv2DWithBatchNorm(const RemapperContext& ctx, int node_index, in FindConv2DWithBatchNorm()
865 const RemapperContext& ctx, int node_index, in FindConv2DWithBatchNormAndActivation()
940 bool FindPadWithConv3D(const RemapperContext& ctx, int node_index, in FindPadWithConv3D()
1014 int node_index, in FindContractionWithBiasAddAndAdd()
1021 const RemapperContext& ctx, int node_index, in FindContractionWithBiasAndAddActivation()
1118 bool FindMatMulBiasAddAndGelu(RemapperContext* ctx, int node_index, in FindMatMulBiasAddAndGelu()
1314 bool FindMulAndMaximum(RemapperContext* ctx, int node_index, in FindMulAndMaximum()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/api/
H A Dprofiler.h157 ScopedOperatorProfile(Profiler* profiler, const char* tag, int node_index) in ScopedOperatorProfile()
165 int node_index) in ScopedDelegateOperatorProfile()
212 #define TFLITE_SCOPED_TAGGED_OPERATOR_PROFILE(profiler, tag, node_index) \ argument
216 #define TFLITE_SCOPED_DELEGATE_OPERATOR_PROFILE(profiler, tag, node_index) \ argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dmodel_builder_test.cc324 TfLiteRegistration** registration) { in TEST()
381 TfLiteRegistration** registration) { in TEST()
444 TfLiteRegistration** registration) { in TEST()
558 TfLiteRegistration** registration) { in TEST()
725 TfLiteRegistration** registration) { in TEST()
943 TfLiteRegistration** registration) { in TEST()
1011 TfLiteRegistration** registration) { in TEST()
1209 TfLiteRegistration** registration) { in TEST()
1282 TfLiteRegistration** registration) { in TEST()
1339 TfLiteRegistration** registration) { in TEST()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dbase_layer_v1.py1388 def get_input_mask_at(self, node_index): argument
1407 def get_output_mask_at(self, node_index): argument
1468 def get_input_shape_at(self, node_index): argument
1487 def get_output_shape_at(self, node_index): argument
1506 def get_input_at(self, node_index): argument
1524 def get_output_at(self, node_index): argument
2028 def _get_node_attribute_at_index(self, node_index, attr, attr_name): argument
H A Dbase_layer.py1897 def get_input_mask_at(self, node_index): argument
1917 def get_output_mask_at(self, node_index): argument
1981 def get_input_shape_at(self, node_index): argument
2001 def get_output_shape_at(self, node_index): argument
2021 def get_input_at(self, node_index): argument
2040 def get_output_at(self, node_index): argument
2589 def _get_node_attribute_at_index(self, node_index, attr, attr_name): argument
/aosp_15_r20/external/tensorflow/tensorflow/lite/profiling/
H A Dprofile_summarizer.cc31 uint32_t node_index; member
73 uint32_t node_index) { in GetOperatorDetails()
125 const auto node_index = event->event_metadata; in ProcessProfiles() local
/aosp_15_r20/external/icing/icing/legacy/index/
H A Dicing-dynamic-trie.h101 Next(uint8_t val, uint32_t node_index) { in Next()
109 uint32_t node_index() const { return used.node_index; } in node_index() function
110 void set_node_index(uint32_t node_index) { used.node_index = node_index; } in set_node_index()
131 uint32_t node_index : 24; member
/aosp_15_r20/hardware/google/pixel/power-libperfmgr/libperfmgr/include/perfmgr/
DNodeLooperThread.h38 : node_index(node_index), in node_index() function
42 std::size_t node_index; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
H A Dcall_test.cc64 int node_index; in BuildCallSubgraph() local
114 int node_index; in BuildGraphWithMultipleOutputs() local
151 int expected_node_index = 0, node_index; in BuildMultiNodeGraph() local
/aosp_15_r20/external/rust/android-crates-io/crates/mls-rs/src/tree_kem/
Dnode.rs255 let node_index = self.validate_index(leaf_index.into())?; in blank_leaf_node() localVariable
293 let node_index = NodeIndex::from(index); in borrow_as_leaf_mut() localVariable
302 let node_index = NodeIndex::from(index); in borrow_as_leaf() localVariable
401 let node_index = (*index as usize) << 1; in insert_leaf() localVariable
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/versioning/
H A Dop_signature_test.cc108 TfLiteNode* node(int node_index) { return &nodes_[node_index]; } in node()
123 TfLiteContext* context, int node_index, TfLiteNode** node, in StubGetNodeAndRegistration()
/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() local
259 for (int node_index : nodes_) { in BuildGraph() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dkernel.cc300 int tensor_index, int node_index) { in ShouldPersistTensorflowTensor()
347 int node_index) { in MaybePersistTensorflowOutputs()
480 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init() local

1234