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.cc52 const NodeDef& weights_node = match.inputs[1].node; in FuseResizePadAndConv() local
101 const NodeDef& weights_node = match.inputs[1].node; in FuseResizeAndConv() local
163 const NodeDef& weights_node = match.inputs[1].node; in FusePadAndConv() local
Dfold_batch_norms.cc56 const NodeDef& weights_node = match.inputs[0].inputs[1].node; in FoldBatchNorms() local
Dfold_old_batch_norms.cc114 const NodeDef& weights_node = conv_node_match.inputs[1].node; in FuseScaleOffsetToConvWeights() local
Dsparsify_gather.cc323 const NodeDef& weights_node = match.inputs[0].inputs[0].node; in SparsifyGatherInternal() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1690 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1740 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local