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::__anon3a6d31fa0111::SymbolicShapesTest
731 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext736 ShapeHandle InferenceContext::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext1310 ShapeHandle InferenceContext::ShapeManager::MakeShape( in MakeShape() function in tensorflow::shape_inference::InferenceContext::ShapeManager
163 Status OpKernel::MakeShape(const Tensor& shape, TensorShape* out) const { in MakeShape() function in tensorflow::OpKernel
940 TEST_F(ShapeInferenceTest, MakeShape) { in TEST_F() argument
158 /* static */ Shape ShapeUtil::MakeShape(PrimitiveType element_type, in MakeShape() function in xla::ShapeUtil163 /* static */ Shape ShapeUtil::MakeShape( in MakeShape() function in xla::ShapeUtil