Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_batch_norms.cc93 auto scaled_weights_vector = scaled_weights.flat<float>(); in FoldBatchNorms() local
95 scaled_weights_vector(row) = in FoldBatchNorms()
Dfold_old_batch_norms.cc130 auto scaled_weights_vector = scaled_weights.flat<float>(); in FuseScaleOffsetToConvWeights() local
132 scaled_weights_vector(row) = in FuseScaleOffsetToConvWeights()