Home
last modified time | relevance | path

Searched refs:ShapesSymbolicallyEqual (Results 1 – 6 of 6) 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.cc76 bool ShapesSymbolicallyEqual(const TensorShapeProto& left, in ShapesSymbolicallyEqual() function
92 bool ShapesSymbolicallyEqual(const OpInfo::TensorProperties& left, in ShapesSymbolicallyEqual() function
94 return ShapesSymbolicallyEqual(left.shape(), right.shape()); in ShapesSymbolicallyEqual()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc889 *shapes_match = ShapesSymbolicallyEqual(*lhs, *rhs); in GetUniqueFactors()
1992 return ShapesSymbolicallyEqual(input_props->shape(), in InputMatchesTargetShape()
2630 ShapesSymbolicallyEqual(value_props.shape(), output_shape)) { in TrySimplify()
2643 ShapesSymbolicallyEqual(value_props.shape(), output_shape) && in TrySimplify()
2749 if (!ShapesSymbolicallyEqual(t.shape(), broadcast_shape)) { in TrySimplifyInternal()
2818 if (!ShapesSymbolicallyEqual(t.shape(), broadcast_shape)) { in TrySimplify()
Dconstant_folding.cc1207 if (!ShapesSymbolicallyEqual(filter_shape, new_filter_shape)) { in IsValidConstShapeForMulConvPushDown()
2528 if (ShapesSymbolicallyEqual(input_props[1], input_props[2]) && in SimplifySelect()
2529 (ShapesSymbolicallyEqual(input_props[0], input_props[1]) || in SimplifySelect()
2948 ShapesSymbolicallyEqual(output_shape, y_shape); in SimplifyArithmeticOperations()
2950 ShapesSymbolicallyEqual(output_shape, x_shape); in SimplifyArithmeticOperations()
3501 if (!ShapesSymbolicallyEqual(mul_shape, conv_shape)) { in MulConvPushDown()
Dremapper.cc660 ShapesSymbolicallyEqual(props[0].shape(), props[1].shape())) { in IsCpuCompatibleDataType()
936 !ShapesSymbolicallyEqual(props[0].shape(), props[1].shape())) in IsCpuCompatibleDataType()