Home
last modified time | relevance | path

Searched defs:input_node (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/tools/graph_transforms/
Dobfuscate_names.cc49 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
74 for (const NodeDef& input_node : input_graph_def.node()) { in ObfuscateNames() local
Dremove_ema.cc115 const NodeDef& input_node = match.inputs[0].node; in RemoveEMA() local
Dfold_batch_norms.cc56 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FoldBatchNorms() local
Dremove_nodes.cc83 const NodeDef& input_node = match.inputs[0].node; in RemoveNodes() local
Dflatten_atrous.cc58 const NodeDef& input_node = match.inputs[0].inputs[0].inputs[0].node; in FlattenAtrousConv() local
Dsparsify_gather_test.cc81 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestSinglePartition() local
295 NodeDef* input_node = CreateNode("ids", "Const", {}, &graph_def); in TestMultiPartition() local
Dfuse_convolutions.cc165 const NodeDef& input_node = match.inputs[0].inputs[0].node; in FusePadAndConv() local
Dtransform_utils.cc343 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch() local
380 for (const NodeDef& input_node : input_graph_def.node()) { in ReplaceMatchingOpTypes() local
556 string input_node = NodeNameFromInput(input); in FindInvalidInputs() local
/external/tensorflow/tensorflow/core/kernels/data/
Dstats_dataset_ops.cc77 Node* input_node; in AsGraphDefInternal() local
173 Node* input_node; in AsGraphDefInternal() local
Dzip_dataset_op.cc85 Node* input_node; in AsGraphDefInternal() local
Ddense_to_sparse_batch_dataset_op.cc120 Node* input_node; in AsGraphDefInternal() local
Dscan_dataset_op.cc112 Node* input_node; in AsGraphDefInternal() local
/external/tensorflow/tensorflow/contrib/receptive_field/python/util/
Dreceptive_field.py159 input_node, argument
/external/tensorflow/tensorflow/tools/quantization/
Dquantize_graph.py241 def quantize_weight_rounded(input_node): argument
265 def quantize_weight_eightbit(input_node, quantization_mode): argument
509 def quantize_node(self, input_node): argument
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc288 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode() local
445 const NodeDef* input_node = node_map_->GetNode(input); in TransitiveReduction() local
Dgraph_rewriter.cc202 const NodeDef* input_node = itr->second->def; in ForwardInputsInternal() local
Dmemory_optimizer.cc88 const NodeDef* input_node = node_map.GetNode(input_name); in FindCandidateRecomputeNodes() local
121 const NodeDef* input_node = node_map.GetNode(input_name_raw); in connected_subgraph() local
785 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
816 const NodeDef* input_node = it1->second; in FindSwapInTrigger() local
Dlayout_optimizer.cc586 auto input_node = node_map_->GetNode(node_->input(input_index)); in UpdateAttrValueOfInput() local
619 auto input_node = node_map_->GetNode(node_->input(pos)); in AddLayoutTransposeToInputs() local
857 auto input_node = node_map_->GetNode(input); in GetOrAddNodePermNHWCToNCHW() local
1164 auto input_node = node_map_->GetNode(node.input(pos)); in IsNodeAfterNCHWToNHWC() local
1183 auto input_node = node_map_->GetNode(current_node->input(pos)); in IsNodeAfterNCHWToNHWC() local
1316 auto input_node = node_map_->GetNode(node_->input(vector_index)); in CustomizedProcessing() local
1480 auto input_node = node_map_->GetNode(input); in IsEveryInputAfterNCHWToNHWC() local
Dauto_parallel.cc178 for (const auto& input_node : input_nodes) { in Initialize() local
Dconstant_folding.cc741 const NodeDef* input_node = node_map_->GetNode(input); in IsFoldable() local
915 const NodeDef* input_node = node_map_->GetNode(input); in EvaluateOneFoldable() local
969 NodeDef* input_node = node_map_->GetNode(input); in FoldNode() local
1060 NodeDef* input_node = node_map_->GetNode(input); in FoldNode() local
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc121 for (const Node* const input_node : node->in_nodes()) { in LoadGraphFromProto() local
305 for (const Node* const input_node : node.in_nodes()) { in AreAllInputsCached() local
344 Node* input_node; in TransformGraphToAddAggregatedInputNode() local
726 const Node* input_node = edge->src(); in RegisterPadNode() local
869 const Node* input_node = edge->src(); in AddNodeInputByInputIndex() local
/external/tensorflow/tensorflow/core/graph/
Dmkl_tfconversion_pass.cc375 Node* input_node = nullptr; in RunPass() local
Dgraph.cc246 Status Node::input_node(int idx, Node** n) const { in input_node() function in tensorflow::Node
257 Status Node::input_node(int idx, const Node** const_n) const { in input_node() function in tensorflow::Node
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc47 NodeDef* input_node = g->add_node(); in ConstructFunctionGraph() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc219 string input_node = ParseNodeName(input, &position); in InferFromTrace() local

12