Searched refs:ShapeAfterBroadcast (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | symbolic_shapes_test.cc | 80 ShapeAfterBroadcast(MakeShape({1, 2}), MakeShape({1, 2}), &output_shape)); in TEST_F()
82 EXPECT_TRUE(ShapeAfterBroadcast(MakeShape({-2, 2}), MakeShape({-2, 2}), in TEST_F()
85 EXPECT_TRUE(ShapeAfterBroadcast(MakeShape({-2, 32}), MakeShape({-2, 1}), in TEST_F()
88 EXPECT_TRUE(ShapeAfterBroadcast(MakeShape({-2, 1}), MakeShape({1, -2}), in TEST_F()
91 EXPECT_TRUE(ShapeAfterBroadcast(MakeShape({-2, 1}), MakeShape({1, -3}), in TEST_F()
95 ShapeAfterBroadcast(MakeShape({-3}), MakeShape({-2, -3}), &output_shape)); in TEST_F()
|
| D | symbolic_shapes.h | 56 bool ShapeAfterBroadcast(const TensorShapeProto& left,
|
| D | symbolic_shapes.cc | 112 bool ShapeAfterBroadcast(const TensorShapeProto& left, in ShapeAfterBroadcast() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | arithmetic_optimizer.cc | 2746 if (!ShapeAfterBroadcast(t.shape(), c.shape(), &broadcast_shape)) { in TrySimplifyInternal()
2815 if (!ShapeAfterBroadcast(t.shape(), c.shape(), &broadcast_shape)) { in TrySimplify()
|
| D | constant_folding.cc | 1203 if (!ShapeAfterBroadcast(filter_shape, mul_const_input_shape, in IsValidConstShapeForMulConvPushDown()
|