Searched refs:scaled_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms.cc | 93 Tensor scaled_weights(DT_FLOAT, weights.shape()); in FoldBatchNorms() local
94 auto scaled_weights_vector = scaled_weights.flat<float>(); in FoldBatchNorms()
106 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FoldBatchNorms()
|
| D | fold_old_batch_norms.cc | 130 Tensor scaled_weights(DT_FLOAT, weights.shape()); in FuseScaleOffsetToConvWeights() local
131 auto scaled_weights_vector = scaled_weights.flat<float>(); in FuseScaleOffsetToConvWeights()
148 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 329 scaled_weights = np.copy(weights)
331 scaled_weights, flags=["multi_index"], op_flags=["readwrite"])
342 scaled_weights, weights.dtype.type, weights.shape)))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2134 NodeDef* scaled_weights = AddEmptyNode(scaled_weights_node_name); in TrySimplify() local
2135 scaled_weights->set_op("Mul"); in TrySimplify()
2136 scaled_weights->set_device(weights->device()); in TrySimplify()
2137 (*scaled_weights->mutable_attr())["T"] = weights->attr().at("dtype"); in TrySimplify()
2138 AddToOptimizationQueue(scaled_weights); in TrySimplify()
2141 scaled_weights->add_input(conv->input(1)); in TrySimplify()
2142 ctx().node_map->AddOutput(weights->name(), scaled_weights->name()); in TrySimplify()
2143 scaled_weights->add_input(mul->input(1)); in TrySimplify()
2144 ctx().node_map->AddOutput(scale->name(), scaled_weights->name()); in TrySimplify()
2145 ForwardControlDependencies(scaled_weights, {source}); in TrySimplify()
[all …]
|