Home
last modified time | relevance | path

Searched refs:ShapesBroadcastable (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes_test.cc64 TEST_F(SymbolicShapesTest, ShapesBroadcastable) { in TEST_F() argument
65 EXPECT_FALSE(ShapesBroadcastable(MakeUnknown(), MakeUnknown())); in TEST_F()
66 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-2}), MakeShape({1, -3}))); in TEST_F()
67 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-1, 2}), MakeShape({-1, 2}))); in TEST_F()
68 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-2, 2}), MakeShape({-3, 2}))); in TEST_F()
69 EXPECT_FALSE(ShapesBroadcastable(MakeShape({-2, 4}), MakeShape({-2, 8}))); in TEST_F()
71 EXPECT_TRUE(ShapesBroadcastable(MakeShape({1, 2}), MakeShape({1, 2}))); in TEST_F()
72 EXPECT_TRUE(ShapesBroadcastable(MakeShape({-2, 2}), MakeShape({-2, 2}))); in TEST_F()
73 EXPECT_TRUE(ShapesBroadcastable(MakeShape({-2, 32}), MakeShape({-2, 1}))); in TEST_F()
74 EXPECT_TRUE(ShapesBroadcastable(MakeShape({-2, 1}), MakeShape({1, -2}))); in TEST_F()
[all …]
Dsymbolic_shapes.h52 bool ShapesBroadcastable(const TensorShapeProto& left,
54 bool ShapesBroadcastable(const OpInfo::TensorProperties& left,
Dsymbolic_shapes.cc97 bool ShapesBroadcastable(const TensorShapeProto& left, in ShapesBroadcastable() function
107 bool ShapesBroadcastable(const OpInfo::TensorProperties& left, in ShapesBroadcastable() function
109 return ShapesBroadcastable(left.shape(), right.shape()); in ShapesBroadcastable()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc410 ShapesBroadcastable(properties, *input_props); in HasAllInputsBroadcastableToShape()