Home
last modified time | relevance | path

Searched defs:input_nodes (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports.cc37 std::vector<NodeDef>* new_nodes) { in BackportConcatV2Transform()
72 const std::set<string>& input_nodes, in BackportTensorArrayV3Transform()
74 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
118 [](const NodeMatch& match, const std::set<string>& input_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()
95 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizeAndConv()
97 std::vector<NodeDef>* new_nodes) { in FuseResizeAndConv()
155 [](const NodeMatch& match, const std::set<string>& input_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()
Dstrip_unused_nodes.cc121 std::set<string> input_nodes; in StripUnusedNodes() local
Dsparsify_gather.cc261 std::vector<NodeDef>* new_nodes) { in SparsifyGatherInternal()
450 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
Dfold_constants_lib.cc54 StringPieceSet input_nodes; in ReplaceSendRecvs() local
Dtransform_utils_test.cc437 std::vector<NodeDef>* new_nodes) { in TestReplaceMatchingOpTypes()
452 {}, &replaced_graph_def)); in TestReplaceMatchingOpTypes()
Dtransform_utils.cc401 std::set<string> input_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
Dgraph_util_test.py30 def GetNewNode(name, op, input_nodes): argument
/external/libaom/libaom/av1/encoder/
Dml.c21 void av1_nn_predict_c(const float *input_nodes, in av1_nn_predict_c()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc463 std::unordered_map<const Node*, std::vector<int>>* input_nodes) in ProcessInputs()
522 const std::unordered_map<const Node*, std::vector<int>>& input_nodes, in ComputeBodyNodes()
572 std::unordered_map<const Node*, std::vector<int>> input_nodes; in TF_GraphToFunctionWithControlOutputs() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc166 const NodeDef& node, const std::vector<NodeDef*>& input_nodes, in BypassingNodeIsBeneficial()
357 std::vector<NodeDef*> input_nodes; in OptimizeNode() local
Dauto_parallel.cc161 std::vector<const NodeDef*> input_nodes; in Initialize() local
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c153 void av1_nn_predict_sse3(const float *input_nodes, in av1_nn_predict_sse3()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc164 std::vector<Node*> input_nodes; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/core/common_runtime/
Dshape_refiner.cc215 std::vector<const Node*> input_nodes(node->num_inputs()); in AddNode() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc410 std::vector<Node*> input_nodes; in CreateTRTNode() local

12