Home
last modified time | relevance | path

Searched refs:InferCollectivePermuteShape (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h151 static StatusOr<Shape> InferCollectivePermuteShape(const Shape& shape);
Dhlo_verifier.cc371 return CheckShape(hlo, ShapeInference::InferCollectivePermuteShape( in HandleCollectivePermute()
Dshape_inference.cc2107 /* static */ StatusOr<Shape> ShapeInference::InferCollectivePermuteShape( in InferCollectivePermuteShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc3004 ShapeInference::InferCollectivePermuteShape(*operand_shape)); in CollectivePermute()