Searched refs:mul_const_input_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 1031 const TensorShapeProto& mul_const_input_shape) { in IsValidConstShapeForMulConvPushDown() argument
1035 if (mul_const_input_shape.dim_size() <= data_format.size() && in IsValidConstShapeForMulConvPushDown()
1036 TensorShape(mul_const_input_shape).num_elements() == 1) { in IsValidConstShapeForMulConvPushDown()
1043 if (!ShapeAfterBroadcast(filter_shape, mul_const_input_shape, in IsValidConstShapeForMulConvPushDown()
1052 for (int i = 0; i < mul_const_input_shape.dim_size() - 1; ++i) { in IsValidConstShapeForMulConvPushDown()
1053 if (mul_const_input_shape.dim(i).size() > 1) return false; in IsValidConstShapeForMulConvPushDown()
|
| D | constant_folding_test.cc | 126 const TensorShape& mul_const_input_shape, in MulConvPushDownTest() argument
156 Tensor mul_const_input(DT_FLOAT, mul_const_input_shape); in MulConvPushDownTest()
365 for (auto mul_const_input_shape : in TEST_F()
370 /*filter_shape=*/{2, 2, 3, 5}, mul_const_input_shape, in TEST_F()
415 for (auto mul_const_input_shape : in TEST_F()
419 /*filter_shape=*/{3, 3, 3, 3}, mul_const_input_shape, in TEST_F()
429 for (auto mul_const_input_shape : in TEST_F()
433 /*filter_shape=*/{3, 3, 3, 3}, mul_const_input_shape, in TEST_F()
|