Home
last modified time | relevance | path

Searched refs:InferGetDimensionSizeShape (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_get_dimension_size_rewriter.cc38 ShapeInference::InferGetDimensionSizeShape( in ReplaceGetSize()
Dshape_inference.h295 static StatusOr<Shape> InferGetDimensionSizeShape(const Shape& shape,
Dhlo_verifier.cc842 ShapeInference::InferGetDimensionSizeShape( in HandleGetDimensionSize()
Dshape_inference.cc2184 /* static */ StatusOr<Shape> ShapeInference::InferGetDimensionSizeShape( in InferGetDimensionSizeShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2498 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferGetDimensionSizeShape( in GetDimensionSize()