Home
last modified time | relevance | path

Searched refs:scaled_weights_node (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms.cc101 NodeDef scaled_weights_node; in FoldBatchNorms() local
102 scaled_weights_node.set_op("Const"); in FoldBatchNorms()
103 scaled_weights_node.set_name(weights_node.name()); in FoldBatchNorms()
104 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FoldBatchNorms()
105 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FoldBatchNorms()
106 new_nodes->push_back(scaled_weights_node); in FoldBatchNorms()
Dfold_old_batch_norms.cc143 NodeDef scaled_weights_node; in FuseScaleOffsetToConvWeights() local
144 scaled_weights_node.set_op("Const"); in FuseScaleOffsetToConvWeights()
145 scaled_weights_node.set_name(weights_node.name()); in FuseScaleOffsetToConvWeights()
146 SetNodeAttr("dtype", DT_FLOAT, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
147 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
148 new_nodes->push_back(scaled_weights_node); in FuseScaleOffsetToConvWeights()