Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports.cc37 std::vector<NodeDef>* new_nodes) { in BackportConcatV2Transform()
73 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform()
74 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
119 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform()
120 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
127 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
Dfuse_convolutions.cc48 std::vector<NodeDef>* new_nodes) { in FuseResizePadAndConv()
96 const std::set<string>& output_nodes, in FuseResizeAndConv()
97 std::vector<NodeDef>* new_nodes) { in FuseResizeAndConv()
156 const std::set<string>& output_nodes, in FusePadAndConv()
157 std::vector<NodeDef>* new_nodes) { in FusePadAndConv()
187 {}, &replaced_graph_def)); in FusePadAndConv()
Dfold_old_batch_norms.cc314 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
319 {}, &replaced_graph_def)); in FoldOldBatchNorms()
351 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
356 {}, &replaced_graph_def)); in FoldOldBatchNorms()
394 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
399 {}, &replaced_graph_def)); in FoldOldBatchNorms()
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()
918 {}, &quantized_graph_def)); in QuantizeNodes()
Dfold_batch_norms.cc51 std::vector<NodeDef>* new_nodes) { in FoldBatchNorms()
118 {}, &replaced_graph_def)); in FoldBatchNorms()
Dremove_nodes.cc73 std::vector<NodeDef>* new_nodes) { in RemoveNodes()
100 {true}, &replaced_graph_def)); in RemoveNodes()
Dround_weights.cc42 std::vector<NodeDef>* new_nodes) { in RoundWeights()
107 {}, output_graph_def)); in RoundWeights()
Dquantize_weights.cc45 std::vector<NodeDef>* new_nodes) { in QuantizeWeights()
138 {}, output_graph_def)); in QuantizeWeights()
Dflatten_atrous.cc52 std::vector<NodeDef>* new_nodes) { in FlattenAtrousConv()
132 {}, &replaced_graph_def)); in FlattenAtrousConv()
Dfuse_quantized_convolution.cc57 std::vector<NodeDef>* new_nodes) { in FuseQuantizedConvolutionAndRequantize()
137 {}, &replaced_graph_def)); in FuseQuantizedConvolutionAndRequantize()
Dsparsify_gather.cc261 std::vector<NodeDef>* new_nodes) { in SparsifyGatherInternal()
450 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
Dtransform_utils_test.cc437 std::vector<NodeDef>* new_nodes) { in TestReplaceMatchingOpTypes()
452 {}, &replaced_graph_def)); in TestReplaceMatchingOpTypes()
Dtransform_utils.cc402 std::set<string> output_nodes; in ReplaceMatchingOpTypes() local
/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dgraph_util.py37 def fuse_op(graph_def, input_nodes, output_nodes, output_dtypes, argument
/external/libaom/libaom/av1/encoder/
Dml.c33 float *output_nodes = buf[buf_index]; in av1_nn_predict_c() local
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h74 absl::InlinedVector<string, 1> output_nodes; member
Dfunctions.cc619 absl::InlinedVector<string, 1> output_nodes; in MakeGrapplerFunctionItem() local
793 absl::flat_hash_set<absl::string_view> output_nodes; in MakeFunctionDef() local
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc193 const std::unordered_set<int>& output_nodes) { in GetStopBackpropNodes()
266 std::unordered_set<int> output_nodes; in Initialize() local
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py196 def _getGraphOpTypes(self, graphdef, output_nodes): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc236 const std::set<NodeDef*> output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode() local
354 const std::vector<NodeDef*> output_nodes(output_node_set.begin(), in OptimizeNode() local
Dfunction_optimizer.cc1041 absl::flat_hash_set<absl::string_view> output_nodes; in InlineDirectFunctionCall() local
1348 absl::flat_hash_set<absl::string_view> output_nodes; in MaybeDeadOutputs() local
1656 absl::flat_hash_set<string> output_nodes; in InlineIndirectFunctionCall() local
Dscoped_allocator_optimizer.cc478 std::set<NodeDef*> output_nodes = node_map->GetOutputs(old_op->name()); in RewireSubgraph() local
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c165 float *const output_nodes = output_layer ? output : buf[buf_index]; in av1_nn_predict_sse3() local
/external/adhd/cras/src/common/
Dcras_types.h384 struct cras_ionode_info output_nodes[CRAS_MAX_IONODES]; member
/external/tensorflow/tensorflow/core/graph/
Dmkl_layout_pass.cc1581 int list_length, std::vector<NodeBuilder::NodeOut>* output_nodes) { in GetNodesProducingTFTensorList()
1646 int list_length, std::vector<NodeBuilder::NodeOut>* output_nodes) { in GetNodesProducingMklTensorList()

12