Home
last modified time | relevance | path

Searched refs:weights_node (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms.cc56 const NodeDef& weights_node = match.inputs[0].inputs[1].node; in FoldBatchNorms() local
60 for (const auto& node : {conv_node, weights_node, mul_values_node}) { in FoldBatchNorms()
64 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
70 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FoldBatchNorms()
104 scaled_weights_node.set_name(weights_node.name()); in FoldBatchNorms()
Dfuse_convolutions.cc52 const NodeDef& weights_node = match.inputs[1].node; in FuseResizePadAndConv() local
57 new_nodes->push_back(weights_node); in FuseResizePadAndConv()
101 const NodeDef& weights_node = match.inputs[1].node; in FuseResizeAndConv() local
104 new_nodes->push_back(weights_node); in FuseResizeAndConv()
163 const NodeDef& weights_node = match.inputs[1].node; in FusePadAndConv() local
168 new_nodes->push_back(weights_node); in FusePadAndConv()
Dsparsify_gather.cc323 const NodeDef& weights_node = match.inputs[0].inputs[0].node; in SparsifyGatherInternal() local
326 TF_RETURN_IF_ERROR(GetNodeAttr(weights_node, "dtype", &data_type)); in SparsifyGatherInternal()
332 weights_node.op() + "' with name '", in SparsifyGatherInternal()
333 weights_node.name(), "' and dtype '", data_type, "'."); in SparsifyGatherInternal()
337 if (weights_node.op() == "Const") { in SparsifyGatherInternal()
338 weight = GetNodeTensorAttr(weights_node, "value"); in SparsifyGatherInternal()
341 weights_node.name(), ckpt_reader, in SparsifyGatherInternal()
342 (*shapes_and_slices)[weights_node.name()], &weight)); in SparsifyGatherInternal()
345 removed_node_names.push_back(weights_node.name()); in SparsifyGatherInternal()
360 StrCat(weights_node.name(), "/indices"), in SparsifyGatherInternal()
[all …]
Dfold_old_batch_norms.cc114 const NodeDef& weights_node = conv_node_match.inputs[1].node; in FuseScaleOffsetToConvWeights() local
115 CHECK_EQ("Const", weights_node.op()); in FuseScaleOffsetToConvWeights()
117 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FuseScaleOffsetToConvWeights()
146 scaled_weights_node.set_name(weights_node.name()); in FuseScaleOffsetToConvWeights()
DREADME.md975 const NodeDef& weights_node = match.inputs[1].node;
978 new_nodes->push_back(weights_node);
/external/tensorflow/tensorflow/contrib/receptive_field/python/util/
Dparse_layer_parameters.py90 weights_node = name_to_node[weights_layer_param_name]
91 if weights_node.op != "VariableV2":
93 shape = weights_node.attr["shape"]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer_test.cc1690 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1696 ASSERT_NE(weights_node, nullptr); in TEST_F()
1703 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()
1740 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1745 ASSERT_NE(weights_node, nullptr); in TEST_F()
1750 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()