Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/utils/
Dsymbolic_shapes.cc41 bool ShapeIsSymbolicallyDefined(const TensorShapeProto& shape) { in ShapeIsSymbolicallyDefined() function
48 bool ShapeIsSymbolicallyDefined(const OpInfo::TensorProperties& properties) { in ShapeIsSymbolicallyDefined() function
49 return ShapeIsSymbolicallyDefined(properties.shape()); in ShapeIsSymbolicallyDefined()
96 if (!ShapeIsSymbolicallyDefined(left) || !ShapeIsSymbolicallyDefined(right)) { in ShapesBroadcastable()
112 if (!ShapeIsSymbolicallyDefined(left) || !ShapeIsSymbolicallyDefined(right)) { in ShapeAfterBroadcast()
Dsymbolic_shapes_test.cc47 TEST_F(SymbolicShapesTest, ShapeIsSymbolicallyDefined) { in TEST_F() argument
48 EXPECT_FALSE(ShapeIsSymbolicallyDefined(MakeUnknown())); in TEST_F()
49 EXPECT_FALSE(ShapeIsSymbolicallyDefined(MakeShape({-1, 2}))); in TEST_F()
51 EXPECT_TRUE(ShapeIsSymbolicallyDefined(MakeShape({1, 2}))); in TEST_F()
52 EXPECT_TRUE(ShapeIsSymbolicallyDefined(MakeShape({-2, 2}))); in TEST_F()
Dsymbolic_shapes.h32 bool ShapeIsSymbolicallyDefined(const TensorShapeProto& shape);
33 bool ShapeIsSymbolicallyDefined(const OpInfo::TensorProperties& properties);
/external/tensorflow/tensorflow/core/grappler/optimizers/
Darithmetic_optimizer.cc471 return has_properties.ok() && ShapeIsSymbolicallyDefined(properties) && in IsSupported()
909 return has_properties.ok() && ShapeIsSymbolicallyDefined(properties) && in IsSupported()