Searched refs:conv_shape (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | transpose_folding_test.cc | 242 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
245 EXPECT_IS_OK(conv_shape); in TEST_F()
247 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
298 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
301 EXPECT_IS_OK(conv_shape); in TEST_F()
303 conv_shape.ValueOrDie(), x, transpose_y, in TEST_F()
359 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
362 EXPECT_IS_OK(conv_shape); in TEST_F()
364 conv_shape.ValueOrDie(), transpose_x, y, in TEST_F()
426 StatusOr<Shape> conv_shape = ShapeInference::InferConvolveShape( in TEST_F() local
[all …]
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 2952 const auto& conv_shape = conv_props[0].shape(); in MulConvPushDown() local
2953 if (!ShapesSymbolicallyEqual(mul_shape, conv_shape)) { in MulConvPushDown()
|