Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h141 static StatusOr<Shape> InferAllToAllShape(const Shape& shape,
Dhlo_verifier.cc322 hlo, ShapeInference::InferAllToAllShape( in HandleAllToAll()
Dshape_inference.cc2061 /* static */ StatusOr<Shape> ShapeInference::InferAllToAllShape( in InferAllToAllShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2878 ShapeInference::InferAllToAllShape(*operand_shape, split_dimension, in AllToAllArray()
2938 ShapeInference::InferAllToAllShape(*operand_shape, split_dimension, in AllToAllTuple()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1208 auto new_shape = ShapeInference::InferAllToAllShape( in ReshardWithAllToAll()