Home
last modified time | relevance | path

Searched refs:InferVariadicOpShape (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h75 static StatusOr<Shape> InferVariadicOpShape(
77 static StatusOr<Shape> InferVariadicOpShape(
Dshape_inference.cc1065 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference
1072 return InferVariadicOpShape(opcode, operand_shapes); in InferVariadicOpShape()
1075 /* static */ StatusOr<Shape> ShapeInference::InferVariadicOpShape( in InferVariadicOpShape() function in xla::ShapeInference
2104 return InferVariadicOpShape(HloOpcode::kTuple, operand_shapes); in InferAllToAllTupleShape()
Dshape_inference_test.cc308 ShapeInference::InferVariadicOpShape(HloOpcode::kTuple, {&s32_, &f32_}); in TEST_F()
2370 ShapeInference::InferVariadicOpShape(HloOpcode::kSort, {&keys, &values}); in TEST_F()
2381 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()
2393 StatusOr<Shape> statusor = ShapeInference::InferVariadicOpShape( in TEST_F()
Dhlo_verifier.cc1271 ShapeInference::InferVariadicOpShape( in CheckVariadicShape()
Dhlo_parser.cc1413 return ShapeInference::InferVariadicOpShape(opcode, arg_shapes); in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc1243 ShapeInference::InferVariadicOpShape( in Tuple()
2068 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferVariadicOpShape( in Sort()