Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference.h299 static StatusOr<Shape> InferCallShape(
Dshape_inference_test.cc2138 ShapeInference::InferCallShape({}, ShapeUtil::MakeProgramShape({}, f32_)); in TEST_F()
2142 auto inferred_status1 = ShapeInference::InferCallShape( in TEST_F()
2150 auto inferred_status_error0 = ShapeInference::InferCallShape( in TEST_F()
2156 auto inferred_status_error1 = ShapeInference::InferCallShape( in TEST_F()
2162 auto inferred_status_error2 = ShapeInference::InferCallShape( in TEST_F()
Dshape_inference.cc3178 /* static */ StatusOr<Shape> ShapeInference::InferCallShape( in InferCallShape() function in xla::ShapeInference
Dhlo_parser.cc1560 return ShapeInference::InferCallShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc833 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferCallShape( in Call()