Searched refs:InferMapShape (Results 1 – 7 of 7) sorted by relevance
945 auto inferred_status = ShapeInference::InferMapShape({&arg}, to_apply, {0}); in TEST_F()952 auto inferred_status_r1f32 = ShapeInference::InferMapShape( in TEST_F()960 auto inferred_status_r1f32_one = ShapeInference::InferMapShape( in TEST_F()966 auto inferred_status_r2s32 = ShapeInference::InferMapShape( in TEST_F()973 auto no_args_error = ShapeInference::InferMapShape( in TEST_F()979 auto args_diff_shapes_error = ShapeInference::InferMapShape( in TEST_F()986 auto arity_error = ShapeInference::InferMapShape( in TEST_F()993 auto output_shape_error = ShapeInference::InferMapShape( in TEST_F()1000 auto param_shape_error = ShapeInference::InferMapShape( in TEST_F()1007 auto param_element_type_error = ShapeInference::InferMapShape( in TEST_F()[all …]
82 static StatusOr<Shape> InferMapShape(
316 ShapeInference::InferMapShape( in MakeMapHlo()
883 ShapeInference::InferMapShape( in HandleMap()
1116 /* static */ StatusOr<Shape> ShapeInference::InferMapShape( in InferMapShape() function in xla::ShapeInference
1802 return ShapeInference::InferMapShape( in ParseInstructionRhs()
2139 Shape shape, ShapeInference::InferMapShape( in Map()