Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/lite/delegates/xnnpack/
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 …]
/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
/external/tensorflow/tensorflow/lite/kernels/
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 …]
Dvariable_ops_test.cc66 int node_index; in ConstructInvalidGraph() local
101 int node_index; in ConstructGraph() local
132 int node_index; in ConstructGraphWithKnownShape() local
/external/tensorflow/tensorflow/lite/
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
Dtensorflow_profiler_logger_shim.cc30 int node_index) {} in OnTfLiteOpPrepare()
47 const char* op_name, int subgraph_index, int node_index) { in OnTfLiteOpInvoke()
Dtensorflow_profiler_logger.cc113 int node_index) { in OnTfLiteOpPrepare()
139 int node_index) { in OnTfLiteOpInvoke()
/external/tensorflow/tensorflow/lite/core/api/
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
/external/tensorflow/tensorflow/lite/core/
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 …]
/external/tensorflow/tensorflow/lite/profiling/
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
/external/tensorflow/tensorflow/core/grappler/optimizers/
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 …]
/external/tensorflow/tensorflow/lite/tools/versioning/
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()
/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/nnapi/
Dnnapi_compatibility_lib.cc44 for (int node_index : TfLiteIntArrayView(execution_plan)) { in CheckCompatibility() local
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util_test.cc43 auto node_index = g.BuildNodeNameIndex(); in TEST() local
/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/lite/experimental/acceleration/mini_benchmark/
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
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
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 …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/
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
/external/icing/icing/legacy/index/
Dicing-dynamic-trie.h99 Next(uint8_t val, uint32_t node_index) { in Next()
107 uint32_t node_index() const { return used.node_index; } in node_index() function
108 void set_node_index(uint32_t node_index) { used.node_index = node_index; } in set_node_index()
129 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.h59 int node_index; member
/external/tensorflow/tensorflow/lite/tools/delegates/compatibility/common/
Ddelegate_compatibility_checker_base.cc77 for (int node_index : TfLiteIntArrayView(execution_plan)) { in checkModelCompatibilityOnline() local

123