Searched refs:InferMapShape (Results 1 – 6 of 6) sorted by relevance
568 auto inferred_status = ShapeInference::InferMapShape({&arg}, to_apply, {0}); in TEST_F()575 auto inferred_status_r1f32 = ShapeInference::InferMapShape( in TEST_F()583 auto inferred_status_r1f32_one = ShapeInference::InferMapShape( in TEST_F()589 auto inferred_status_r2s32 = ShapeInference::InferMapShape( in TEST_F()596 auto no_args_error = ShapeInference::InferMapShape( in TEST_F()602 auto args_diff_shapes_error = ShapeInference::InferMapShape( in TEST_F()609 auto arity_error = ShapeInference::InferMapShape( in TEST_F()616 auto output_shape_error = ShapeInference::InferMapShape( in TEST_F()623 auto param_shape_error = ShapeInference::InferMapShape( in TEST_F()630 auto param_element_type_error = ShapeInference::InferMapShape( in TEST_F()[all …]
82 static StatusOr<Shape> InferMapShape(
244 ShapeInference::InferMapShape( in MakeMapHlo()
613 ShapeInference::InferMapShape( in HandleMap()
1072 /* static */ StatusOr<Shape> ShapeInference::InferMapShape( in InferMapShape() function in xla::ShapeInference
1754 Shape shape, ShapeInference::InferMapShape( in Map()