Searched refs:weights_cols (Results 1 – 2 of 2) sorted by relevance
74 int64 weights_cols; in FoldBatchNorms() local76 weights_cols = weights.shape().dim_size(3); in FoldBatchNorms()78 weights_cols = in FoldBatchNorms()81 weights_cols = weights.shape().dim_size(1); in FoldBatchNorms()84 (mul_values.shape().dim_size(0) != weights_cols)) { in FoldBatchNorms()97 mul_values.flat<float>()(row % weights_cols); in FoldBatchNorms()
117 int64 weights_cols; in FuseScaleOffsetToConvWeights() local119 weights_cols = weights.shape().dim_size(3); in FuseScaleOffsetToConvWeights()121 weights_cols = weights.shape().dim_size(2) * weights.shape().dim_size(3); in FuseScaleOffsetToConvWeights()123 weights_cols = weights.shape().dim_size(1); in FuseScaleOffsetToConvWeights()125 CHECK_EQ(weights_cols, scale_values.size()); in FuseScaleOffsetToConvWeights()133 weights_vector(row) * scale_values[row % weights_cols]; in FuseScaleOffsetToConvWeights()136 Tensor bias_offset(DT_FLOAT, {weights_cols}); in FuseScaleOffsetToConvWeights()138 for (int64 col = 0; col < weights_cols; ++col) { in FuseScaleOffsetToConvWeights()