Searched refs:weights_node (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms.cc | 55 const NodeDef& weights_node = match.inputs[0].inputs[1].node; in FoldBatchNorms() local
59 for (const auto& node : {conv_node, weights_node, mul_values_node}) { in FoldBatchNorms()
63 {mul_node, conv_node, input_node, weights_node, in FoldBatchNorms()
69 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FoldBatchNorms()
103 scaled_weights_node.set_name(weights_node.name()); in FoldBatchNorms()
|
| D | fuse_convolutions.cc | 51 const NodeDef& weights_node = match.inputs[1].node; in FuseResizePadAndConv() local
56 new_nodes->push_back(weights_node); in FuseResizePadAndConv()
100 const NodeDef& weights_node = match.inputs[1].node; in FuseResizeAndConv() local
103 new_nodes->push_back(weights_node); in FuseResizeAndConv()
162 const NodeDef& weights_node = match.inputs[1].node; in FusePadAndConv() local
167 new_nodes->push_back(weights_node); in FusePadAndConv()
|
| D | sparsify_gather.cc | 322 const NodeDef& weights_node = match.inputs[0].inputs[0].node; in SparsifyGatherInternal() local
325 TF_RETURN_IF_ERROR(GetNodeAttr(weights_node, "dtype", &data_type)); in SparsifyGatherInternal()
331 weights_node.op() + "' with name '", in SparsifyGatherInternal()
332 weights_node.name(), "' and dtype '", data_type, "'."); in SparsifyGatherInternal()
336 if (weights_node.op() == "Const") { in SparsifyGatherInternal()
337 weight = GetNodeTensorAttr(weights_node, "value"); in SparsifyGatherInternal()
340 weights_node.name(), ckpt_reader, in SparsifyGatherInternal()
341 (*shapes_and_slices)[weights_node.name()], &weight)); in SparsifyGatherInternal()
344 removed_node_names.push_back(weights_node.name()); in SparsifyGatherInternal()
359 StrCat(weights_node.name(), "/indices"), in SparsifyGatherInternal()
[all …]
|
| D | fold_old_batch_norms.cc | 113 const NodeDef& weights_node = conv_node_match.inputs[1].node; in FuseScaleOffsetToConvWeights() local
114 CHECK_EQ("Const", weights_node.op()); in FuseScaleOffsetToConvWeights()
116 Tensor weights = GetNodeTensorAttr(weights_node, "value"); in FuseScaleOffsetToConvWeights()
145 scaled_weights_node.set_name(weights_node.name()); in FuseScaleOffsetToConvWeights()
|
| D | README.md | 975 const NodeDef& weights_node = match.inputs[1].node;
978 new_nodes->push_back(weights_node);
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 379 for i, weights_node in enumerate(conv_op.input):
380 if weights_node == weights_op.name:
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer_test.cc | 1718 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1724 ASSERT_NE(weights_node, nullptr); in TEST_F()
1734 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()
1771 const NodeDef* weights_node = node_map.GetNode(optimized_weights); in TEST_F() local
1776 ASSERT_NE(weights_node, nullptr); in TEST_F()
1783 EXPECT_EQ(conv_node->input(1), weights_node->name()); in TEST_F()
|