Home
last modified time | relevance | path

Searched refs:MakeValidatedShape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc104 ShapeUtil::MakeValidatedShape(element_type, dimensions)); in MakeShapeWithLayoutInternal()
160 return MakeValidatedShape(element_type, dimensions).ValueOrDie(); in MakeShape()
166 return MakeValidatedShape(element_type, dimensions, dynamic_dimensions) in MakeShape()
170 /* static */ StatusOr<Shape> ShapeUtil::MakeValidatedShape( in MakeValidatedShape() function in xla::ShapeUtil
178 /* static */ StatusOr<Shape> ShapeUtil::MakeValidatedShape( in MakeValidatedShape() function in xla::ShapeUtil
182 MakeValidatedShape(element_type, dimensions)); in MakeValidatedShape()
Dshape_util.h383 static StatusOr<Shape> MakeValidatedShape(PrimitiveType element_type,
385 static StatusOr<Shape> MakeValidatedShape(
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc219 return ShapeUtil::MakeValidatedShape(element_type, output_dimensions, in InferWindowOutputShape()