Home
last modified time | relevance | path

Searched defs:weights_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfuse_convolutions.cc51 const NodeDef& weights_node = match.inputs[1].node; in FuseResizePadAndConv() local
100 const NodeDef& weights_node = match.inputs[1].node; in FuseResizeAndConv() local
162 const NodeDef& weights_node = match.inputs[1].node; in FusePadAndConv() local
Dfold_batch_norms.cc55 const NodeDef& weights_node = match.inputs[0].inputs[1].node; in FoldBatchNorms() local
Dfold_old_batch_norms.cc113 const NodeDef& weights_node = conv_node_match.inputs[1].node; in FuseScaleOffsetToConvWeights() local
Dsparsify_gather.cc322 const NodeDef& weights_node = match.inputs[0].inputs[0].node; in SparsifyGatherInternal() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1718 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1771 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local