Searched refs:scaled_weights (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_batch_norms.cc | 92 Tensor scaled_weights(DT_FLOAT, weights.shape()); in FoldBatchNorms() local
93 auto scaled_weights_vector = scaled_weights.flat<float>(); in FoldBatchNorms()
105 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FoldBatchNorms()
|
| D | fold_old_batch_norms.cc | 129 Tensor scaled_weights(DT_FLOAT, weights.shape()); in FuseScaleOffsetToConvWeights() local
130 auto scaled_weights_vector = scaled_weights.flat<float>(); in FuseScaleOffsetToConvWeights()
147 SetNodeTensorAttr<float>("value", scaled_weights, &scaled_weights_node); in FuseScaleOffsetToConvWeights()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 356 scaled_weights = np.copy(weights)
358 scaled_weights, flags=["multi_index"], op_flags=["readwrite"])
377 scaled_weights, weights.dtype.type, weights.shape)))
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2245 NodeDef* scaled_weights = AddEmptyNode(scaled_weights_node_name); in TrySimplify() local
2246 scaled_weights->set_op(source->op()); in TrySimplify()
2247 scaled_weights->set_device(weights->device()); in TrySimplify()
2248 (*scaled_weights->mutable_attr())["T"] = weights->attr().at("dtype"); in TrySimplify()
2249 AddToOptimizationQueue(scaled_weights); in TrySimplify()
2252 scaled_weights->add_input(conv->input(1)); in TrySimplify()
2253 ctx().node_map->AddOutput(weights->name(), scaled_weights->name()); in TrySimplify()
2254 scaled_weights->add_input(mul->input(scale_idx)); in TrySimplify()
2255 ctx().node_map->AddOutput(scale->name(), scaled_weights->name()); in TrySimplify()
2256 ForwardControlDependencies(scaled_weights, {source}); in TrySimplify()
[all …]
|