Home
last modified time | relevance | path

Searched defs:output_nodes (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dbackports.cc36 std::vector<NodeDef>* new_nodes) { in BackportConcatV2Transform()
72 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform()
73 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
118 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform()
119 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
126 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
H A Dfuse_convolutions.cc47 std::vector<NodeDef>* new_nodes) { in FuseResizePadAndConv()
95 const std::set<string>& output_nodes, in FuseResizeAndConv()
96 std::vector<NodeDef>* new_nodes) { in FuseResizeAndConv()
155 const std::set<string>& output_nodes, in FusePadAndConv()
156 std::vector<NodeDef>* new_nodes) { in FusePadAndConv()
186 {}, &replaced_graph_def)); in FusePadAndConv()
H A Dquantize_nodes.cc282 std::vector<NodeDef>* new_nodes) { in RemoveRedundantQuantizations()
302 {true}, &replaced_graph_def)); in RemoveRedundantQuantizations()
405 std::vector<NodeDef>* new_nodes) { in ConvertFakeQuantsToRequantize()
452 {}, output_graph_def)); in ConvertFakeQuantsToRequantize()
499 std::vector<NodeDef>* new_nodes) { in MergeAdjacentRequantizes()
522 {}, output_graph_def)); in MergeAdjacentRequantizes()
549 std::vector<NodeDef>* new_nodes) { in HoistFakeQuants()
578 {}, &hoisted_graph_def)); in HoistFakeQuants()
698 std::vector<NodeDef>* new_nodes) { in QuantizeNodes()
923 {}, &quantized_graph_def)); in QuantizeNodes()
H A Dfold_old_batch_norms.cc313 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
318 {}, &replaced_graph_def)); in FoldOldBatchNorms()
350 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
355 {}, &replaced_graph_def)); in FoldOldBatchNorms()
393 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
398 {}, &replaced_graph_def)); in FoldOldBatchNorms()
H A Dfold_batch_norms.cc50 std::vector<NodeDef>* new_nodes) { in FoldBatchNorms()
117 {}, &replaced_graph_def)); in FoldBatchNorms()
H A Dremove_nodes.cc72 std::vector<NodeDef>* new_nodes) { in RemoveNodes()
99 {true}, &replaced_graph_def)); in RemoveNodes()
H A Dround_weights.cc42 std::vector<NodeDef>* new_nodes) { in RoundWeights()
108 {}, output_graph_def)); in RoundWeights()
H A Dquantize_weights.cc45 std::vector<NodeDef>* new_nodes) { in QuantizeWeights()
138 {}, output_graph_def)); in QuantizeWeights()
H A Dflatten_atrous.cc52 std::vector<NodeDef>* new_nodes) { in FlattenAtrousConv()
132 {}, &replaced_graph_def)); in FlattenAtrousConv()
H A Dsparsify_gather.cc260 std::vector<NodeDef>* new_nodes) { in SparsifyGatherInternal()
449 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
H A Dtransform_utils_test.cc437 std::vector<NodeDef>* new_nodes) { in TestReplaceMatchingOpTypes()
452 {}, &replaced_graph_def)); in TestReplaceMatchingOpTypes()
H A Dtransform_utils.cc404 std::set<string> output_nodes; in ReplaceMatchingOpTypes() local
/aosp_15_r20/external/libaom/av1/encoder/arm/
H A Dml_neon.c35 float *const output_nodes, bool output_layer) { in nn_propagate_8to1()
65 float *const output_nodes) { in nn_propagate_xto1()
100 float *const output_nodes) { in nn_propagate_xsto1()
119 float *const output_nodes, bool output_layer) { in nn_propagate_4to1()
145 float *const output_nodes, bool output_layer) { in nn_propagate_4to4()
183 float *const output_nodes, bool output_layer) { in nn_propagate_4to8()
228 float *const output_nodes, bool output_layer) { in nn_propagate_8to4()
278 float *const output_nodes = output_layer ? output : buf[buf_index]; in av1_nn_predict_neon() local
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dml_avx2.c31 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to1()
59 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to4()
89 int num_outputs, float *const output_nodes, int is_clip_required) { in nn_propagate_8to8()
120 bool is_output_layer, int num_outputs, float *const output_nodes) { in nn_propagate_input_multiple_of_8()
152 float *const output_nodes = is_output_layer ? output : &buf[buf_index][0]; in av1_nn_predict_avx2() local
H A Dml_sse3.c165 float *const output_nodes = output_layer ? output : &buf[buf_index][0]; in av1_nn_predict_sse3() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dgraph_partition.cc63 absl::flat_hash_map<std::string, NodeInfo> output_nodes; member
77 absl::flat_hash_map<std::string, NodeInfo>& output_nodes, in PrepareSubgraphForFunctionConversion()
217 const absl::flat_hash_map<std::string, NodeInfo>& output_nodes, in BuildPartitionedCallOp()
461 absl::flat_hash_map<std::string, NodeInfo>& output_nodes = in InsertTransferOps() local
H A Dtfrt_graph_execution_state.cc266 absl::flat_hash_set<Node*> output_nodes; in PlaceInputOutputNodesOnHost() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Ddependency_optimizer.cc266 const auto output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode() local
395 const std::vector<NodeDef*> output_nodes(output_node_set.begin(), in OptimizeNode() local
/aosp_15_r20/external/pytorch/torch/fx/passes/utils/
H A Dsource_matcher_utils.py44 output_nodes: List[Node] = field(default_factory=list) variable in SourcePartition
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/
H A Dsaved_model.cc723 std::vector<std::string> output_nodes; member
853 const std::vector<std::string>& output_nodes, in ImportSubgraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/
H A Dgraph_executor.h101 std::vector<std::string> output_nodes; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/
H A Dconvert_test.py172 def _getGraphOpTypes(self, graphdef, output_nodes): argument
/aosp_15_r20/system/libufdt/tests/src/
H A Dufdt_gen_test_dts.c114 static void output_nodes(FILE *fp, int count, const char *prop, const char *value) { in output_nodes() function
/aosp_15_r20/external/libaom/av1/encoder/
H A Dml.c44 float *output_nodes = buf[buf_index]; in av1_nn_predict_c() local
/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/
H A Dgradients.cc267 std::unordered_set<int> output_nodes; in Initialize() local

12