Searched defs:MakeShape (Results 1 – 6 of 6) sorted by relevance
42 func MakeShape(shape ...int64) Shape { func
32 TensorShapeProto MakeShape(std::vector<int> dims) { in MakeShape() function in tensorflow::grappler::__anonabcc5d180111::SymbolicShapesTest
673 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext678 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext1285 ShapeHandle InferenceContext::ShapeManager::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext::ShapeManager
372 Status MakeShape(const Tensor& shape, TensorShape* out) { in MakeShape() function
955 TEST_F(ShapeInferenceTest, MakeShape) { in TEST_F() argument
252 /* static */ Shape ShapeUtil::MakeShape(PrimitiveType element_type, in MakeShape() function in xla::ShapeUtil263 /* static */ Shape ShapeUtil::MakeShape( in MakeShape() function in xla::ShapeUtil