Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h133 static StatusOr<Shape> InferAllToAllShape(const Shape& shape,
Dshape_inference.cc1991 /* static */ StatusOr<Shape> ShapeInference::InferAllToAllShape( in InferAllToAllShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2173 ShapeInference::InferAllToAllShape(operand_shape, split_dimension, in AllToAll()