Searched refs:mul_const_input_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1190 const TensorShapeProto& mul_const_input_shape) { in IsValidConstShapeForMulConvPushDown() argument
1194 if (mul_const_input_shape.dim_size() <= in IsValidConstShapeForMulConvPushDown()
1196 TensorShape(mul_const_input_shape).num_elements() == 1) { in IsValidConstShapeForMulConvPushDown()
1203 if (!ShapeAfterBroadcast(filter_shape, mul_const_input_shape, in IsValidConstShapeForMulConvPushDown()
1212 for (int i = 0; i < mul_const_input_shape.dim_size() - 1; ++i) { in IsValidConstShapeForMulConvPushDown()
1213 if (mul_const_input_shape.dim(i).size() > 1) return false; in IsValidConstShapeForMulConvPushDown()
|
| D | constant_folding_test.cc | 132 const TensorShape& mul_const_input_shape, in MulConvPushDownTest() argument
162 Tensor mul_const_input(DT_FLOAT, mul_const_input_shape); in MulConvPushDownTest()
636 for (auto mul_const_input_shape : in TEST_F()
641 /*filter_shape=*/{2, 2, 3, 5}, mul_const_input_shape, in TEST_F()
687 for (auto mul_const_input_shape : in TEST_F()
691 /*filter_shape=*/{3, 3, 3, 3}, mul_const_input_shape, in TEST_F()
701 for (auto mul_const_input_shape : in TEST_F()
705 /*filter_shape=*/{3, 3, 3, 3}, mul_const_input_shape, in TEST_F()
|