Searched refs:mul_values (Results 1 – 2 of 2) sorted by relevance
70 Tensor mul_values = GetNodeTensorAttr(mul_values_node, "value"); in FoldBatchNorms() local83 if ((mul_values.shape().dims() != 1) || in FoldBatchNorms()84 (mul_values.shape().dim_size(0) != weights_cols)) { in FoldBatchNorms()87 mul_values.shape().DebugString()); in FoldBatchNorms()97 mul_values.flat<float>()(row % weights_cols); in FoldBatchNorms()
1136 mul_values = [1139 self.assertAllClose(mul_values, [1.0, 2.0, 4.0, 8.0])1338 mul_values = [1341 self.assertAllClose(mul_values, [6.0, 6.0, 6.0, 6.0])