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()
176 return MakeValidatedShape(element_type, dimensions).ValueOrDie(); in MakeShape()
186 return MakeValidatedShape(element_type, dimensions, dynamic_dimensions) in MakeShape()
197 /* static */ StatusOr<Shape> ShapeUtil::MakeValidatedShape( in MakeValidatedShape() function in xla::ShapeUtil
205 /* static */ StatusOr<Shape> ShapeUtil::MakeValidatedShape( in MakeValidatedShape() function in xla::ShapeUtil
209 MakeValidatedShape(element_type, dimensions)); in MakeValidatedShape()
Dshape_util.h396 static StatusOr<Shape> MakeValidatedShape(PrimitiveType element_type,
398 static StatusOr<Shape> MakeValidatedShape(
/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.cc219 return ShapeUtil::MakeValidatedShape(element_type, output_dimensions, in InferWindowOutputShape()