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::__anonbe1d059d0111::SymbolicShapesTest
680 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext685 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext1292 ShapeHandle InferenceContext::ShapeManager::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext::ShapeManager
397 Status MakeShape(const Tensor& shape, TensorShape* out) { in MakeShape() function
970 TEST_F(ShapeInferenceTest, MakeShape) { in TEST_F() argument
254 /* static */ Shape ShapeUtil::MakeShape(PrimitiveType element_type, in MakeShape() function in xla::ShapeUtil265 /* static */ Shape ShapeUtil::MakeShape( in MakeShape() function in xla::ShapeUtil