Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Dxnnpack_delegate.cc152 const int node_index = params->nodes_to_replace->data[i]; in Create() local
499 const int node_index = params->nodes_to_replace->data[i]; in Create() local
579 int node_index) { in CalculatePadding()
597 TfLiteContext* context, int node_index, TfLiteFusedActivation activation, in ConvertActivationToOutputRange()
641 int node_index) { in CheckConvolutionParams()
671 int output_channels, int node_index) { in CheckDepthwiseConvolutionParams()
716 int node_index) { in CheckMediaPipeTransposedConvolutionParams()
733 int node_index) { in CheckMediaPipePoolParams()
812 int node_index) { in CheckFullyConnectedParams()
825 int node_index) { in CheckPoolingParams()
[all …]
/external/tensorflow/tensorflow/core/grappler/utils/
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()
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
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
/external/tensorflow/tensorflow/lite/kernels/
Dsubgraph_test_util.cc114 int node_index; in BuildAddSubgraph() local
145 int node_index; in BuildMulSubgraph() local
175 int node_index; in BuildPadSubgraph() local
210 int node_index; in BuildIfSubgraph() local
243 int node_index; in BuildLessEqualCondSubgraph() local
278 int node_index; in BuildAccumulateLoopBodySubgraph() local
330 int node_index; in BuildPadLoopBodySubgraph() local
378 int node_index; in BuildWhileSubgraph() local
402 int node_index; in BuildAssignRandomValueToVariableSubgraph() local
433 int node_index; in BuildCallOnceAndReadVariableSubgraph() local
[all …]
Dvariable_ops_test.cc66 int node_index; in ConstructInvalidGraph() local
101 int node_index; in ConstructGraph() local
Dhashtable_ops_test.cc118 int node_index; in BuildDefaultGraph() local
145 int node_index; in BuildNoImportGraph() local
167 int node_index; in BuildImportTwiceGraph() local
199 int node_index; in BuildTwoHashtablesGraph() local
/external/tensorflow/tensorflow/lite/
Dgraph_info.cc46 for (auto node_index : TfLiteIntArrayView(nodes_to_partition)) { in PartitionGraphIntoIndependentNodeSubsetsImpl() local
77 for (int node_index = 0; node_index < info_->num_execution_nodes(); in Partition() local
137 bool UpdateNode(int node_index) { in UpdateNode()
209 for (int node_index = 0; node_index < info_->num_execution_nodes(); in BuildNodeSubset() local
Doptional_debug_tools.cc404 for (size_t node_index = 0; node_index < subgraph.nodes_size(); in PrintInterpreterState() local
420 for (size_t node_index = 0; node_index < subgraph.nodes_size(); in PrintInterpreterState() local
Dinterpreter.h262 int node_index) const { in node_and_registration()
269 int subgraph_index, int node_index) const { in node_and_registration()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.cc76 int node_index, const char* message) { in ReportOpError()
184 int node_index = subgraph_->execution_plan()[index]; in node() local
187 size_t node_index(size_t index) const override { in node_index() function in tflite::InterpreterInfo
242 for (int node_index = 0; node_index < nodes_and_registration_.size(); in ~Subgraph() local
258 void Subgraph::CleanupNode(int node_index) { in CleanupNode()
426 int node_index; in ReplaceNodeSubsetsWithDelegateKernels() local
784 const TfLiteRegistration* registration, int* node_index) { in AddNodeWithParameters()
987 int node_index = execution_plan[execution_plan_index]; in PrepareOpsStartingAt() local
1106 int node_index = execution_plan_[execution_plan_index]; in Invoke() local
1261 int node_index, TfLiteNode** node, TfLiteRegistration** registration) { in GetNodeAndRegistration()
[all …]
/external/tensorflow/tensorflow/lite/core/api/
Dprofiler.h127 ScopedOperatorProfile(Profiler* profiler, const char* tag, int node_index) in ScopedOperatorProfile()
135 int node_index) in ScopedDelegateOperatorProfile()
175 #define TFLITE_SCOPED_TAGGED_OPERATOR_PROFILE(profiler, tag, node_index) \ argument
179 #define TFLITE_SCOPED_DELEGATE_OPERATOR_PROFILE(profiler, tag, node_index) \ argument
/external/tensorflow/tensorflow/lite/profiling/
Dprofile_summarizer.cc30 uint32_t node_index; member
72 uint32_t node_index) { in GetOperatorDetails()
141 const auto node_index = event->event_metadata; in ProcessProfiles() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc403 bool FindContractionWithBias(const RemapperContext& ctx, int node_index, in FindContractionWithBias()
444 const RemapperContext& ctx, int node_index, in FindContractionWithBiasAndActivation()
494 bool FindConv2DWithSqueezeAndBias(const RemapperContext& ctx, int node_index, in FindConv2DWithSqueezeAndBias()
549 bool FindConv2DWithBatchNorm(const RemapperContext& ctx, int node_index, in FindConv2DWithBatchNorm()
598 const RemapperContext& ctx, int node_index, in FindConv2DWithBatchNormAndActivation()
712 int node_index, in FindContractionWithBiasAddAndAdd()
719 const RemapperContext& ctx, int node_index, in FindContractionWithBiasAndAddActivation()
770 bool FindFusedBatchNorm(const RemapperContext& ctx, int node_index, in FindFusedBatchNorm()
829 bool FindFusedBatchNormEx(const RemapperContext& ctx, int node_index, in FindFusedBatchNormEx()
967 bool FindTensorToHashBucket(const RemapperContext& ctx, int node_index, in FindTensorToHashBucket()
[all …]
/external/tensorflow/tensorflow/python/keras/engine/
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
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
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util_test.cc43 auto node_index = g.BuildNodeNameIndex(); in TEST() local
/external/tensorflow/tensorflow/lite/experimental/acceleration/mini_benchmark/
Dcall_test.cc62 int node_index; in BuildCallSubgraph() local
112 int node_index; in BuildGraphWithMultipleOutputs() local
149 int expected_node_index = 0, node_index; in BuildMultiNodeGraph() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dmodel_builder_test.cc320 TfLiteRegistration** registration) { in TEST()
377 TfLiteRegistration** registration) { in TEST()
440 TfLiteRegistration** registration) { in TEST()
554 TfLiteRegistration** registration) { in TEST()
721 TfLiteRegistration** registration) { in TEST()
941 TfLiteRegistration** registration) { in TEST()
1009 TfLiteRegistration** registration) { in TEST()
1207 TfLiteRegistration** registration) { in TEST()
1333 TfLiteNode* node(int node_index) { return &nodes_[node_index]; } in node()
1348 TfLiteContext* context, int node_index, TfLiteNode** node, in StubGetNodeAndRegistration()
/external/tensorflow/tensorflow/lite/delegates/hexagon/
Dhexagon_delegate_kernel.cc80 for (auto node_index : TfLiteIntArrayView(params->nodes_to_replace)) { in Init() local
255 for (int node_index : nodes_) { in BuildGraph() local
/external/icing/icing/legacy/index/
Dicing-dynamic-trie.h97 Next(uint8_t val, uint32_t node_index) { in Next()
105 uint32_t node_index() const { return used.node_index; } in node_index() function
106 void set_node_index(uint32_t node_index) { used.node_index = node_index; } in set_node_index()
127 uint32_t node_index : 24; member
/external/google-breakpad/src/processor/
Dstatic_map-inl.h133 int node_index = 0; in ValidateInMemoryStructure() local
/external/arm-trusted-firmware/lib/psci/
Dpsci_setup.c127 unsigned int node_index = 0U, num_children; in populate_power_domain_tree() local
/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
Dcalibration_common.h58 int node_index; member
/external/tensorflow/tensorflow/python/keras/utils/
Dlayer_utils.py28 def get_source_inputs(tensor, layer=None, node_index=None): argument

123