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::__anon8704cdbc0111::SymbolicShapesTest
671 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext676 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext1250 ShapeHandle InferenceContext::ShapeManager::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext::ShapeManager
370 Status MakeShape(const Tensor& shape, TensorShape* out) { in MakeShape() function
955 TEST_F(ShapeInferenceTest, MakeShape) { in TEST_F() argument
174 /* static */ Shape ShapeUtil::MakeShape(PrimitiveType element_type, in MakeShape() function in xla::ShapeUtil183 /* static */ Shape ShapeUtil::MakeShape( in MakeShape() function in xla::ShapeUtil