Home
last modified time | relevance | path

Searched refs:InferMapShape (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dshape_inference_test.cc945 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 …]
Dshape_inference.h82 static StatusOr<Shape> InferMapShape(
Dhlo_creation_utils.cc316 ShapeInference::InferMapShape( in MakeMapHlo()
Dhlo_verifier.cc883 ShapeInference::InferMapShape( in HandleMap()
Dshape_inference.cc1116 /* static */ StatusOr<Shape> ShapeInference::InferMapShape( in InferMapShape() function in xla::ShapeInference
Dhlo_parser.cc1802 return ShapeInference::InferMapShape( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2139 Shape shape, ShapeInference::InferMapShape( in Map()