Home
last modified time | relevance | path

Searched refs:ShapesSymbolicallyEqual (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes_test.cc55 TEST_F(SymbolicShapesTest, ShapesSymbolicallyEqual) { in TEST_F() argument
56 EXPECT_FALSE(ShapesSymbolicallyEqual(MakeUnknown(), MakeUnknown())); in TEST_F()
57 EXPECT_FALSE(ShapesSymbolicallyEqual(MakeShape({-1, 2}), MakeShape({-1, 2}))); in TEST_F()
58 EXPECT_FALSE(ShapesSymbolicallyEqual(MakeShape({-2, 2}), MakeShape({-3, 2}))); in TEST_F()
60 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({1, 2}), MakeShape({1, 2}))); in TEST_F()
61 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({-2, 2}), MakeShape({-2, 2}))); in TEST_F()
81 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({1, 2}), output_shape)); in TEST_F()
84 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({-2, 2}), output_shape)); in TEST_F()
87 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({-2, 32}), output_shape)); in TEST_F()
90 EXPECT_TRUE(ShapesSymbolicallyEqual(MakeShape({-2, -2}), output_shape)); in TEST_F()
[all …]
Dsymbolic_shapes.h45 bool ShapesSymbolicallyEqual(const TensorShapeProto& left,
47 bool ShapesSymbolicallyEqual(const OpInfo::TensorProperties& left,
Dsymbolic_shapes.cc73 bool ShapesSymbolicallyEqual(const TensorShapeProto& left, in ShapesSymbolicallyEqual() function
89 bool ShapesSymbolicallyEqual(const OpInfo::TensorProperties& left, in ShapesSymbolicallyEqual() function
91 return ShapesSymbolicallyEqual(left.shape(), right.shape()); in ShapesSymbolicallyEqual()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc862 *shapes_match = ShapesSymbolicallyEqual(lhs, rhs); in GetUniqueFactors()
1891 return ShapesSymbolicallyEqual(input_props.shape(), reshape_props.shape()); in ReshapeIsIdentity()
2510 ShapesSymbolicallyEqual(value_props.shape(), output_shape)) { in TrySimplify()
2523 ShapesSymbolicallyEqual(value_props.shape(), output_shape)) { in TrySimplify()
2641 if (!ShapesSymbolicallyEqual(t.shape(), broadcast_shape)) { in TrySimplifyInternal()
2727 if (!ShapesSymbolicallyEqual(t.shape(), broadcast_shape)) { in TrySimplify()
Dconstant_folding.cc1047 if (!ShapesSymbolicallyEqual(filter_shape, new_filter_shape)) { in IsValidConstShapeForMulConvPushDown()
2668 ShapesSymbolicallyEqual(output_shape, y_shape); in SimplifyArithmeticOperations()
2700 ShapesSymbolicallyEqual(output_shape, x_shape); in SimplifyArithmeticOperations()
2953 if (!ShapesSymbolicallyEqual(mul_shape, conv_shape)) { in MulConvPushDown()