Searched refs:weights_vector (Results 1 – 2 of 2) sorted by relevance
91 auto weights_vector = weights.flat<float>(); in FoldBatchNorms() local94 for (int64 row = 0; row < weights_vector.dimension(0); ++row) { in FoldBatchNorms()96 weights_vector(row) * in FoldBatchNorms()
128 auto weights_vector = weights.flat<float>(); in FuseScaleOffsetToConvWeights() local131 for (int64 row = 0; row < weights_vector.dimension(0); ++row) { in FuseScaleOffsetToConvWeights()133 weights_vector(row) * scale_values[row % weights_cols]; in FuseScaleOffsetToConvWeights()