Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h136 static StatusOr<Shape> InferAllReduceShape(
Dhlo_verifier.cc288 return CheckShape(hlo, ShapeInference::InferAllReduceShape(operand_shapes)); in HandleAllReduce()
Dshape_inference.cc2045 /* static */ StatusOr<Shape> ShapeInference::InferAllReduceShape( in InferAllReduceShape() function in xla::ShapeInference
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2810 ShapeInference::InferAllReduceShape(operand_shapes)); in AllReduce()