Searched refs:MakeProgramShape (Results 1 – 5 of 5) sorted by relevance
63 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()88 select_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()90 scatter_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()317 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F()348 ShapeUtil::MakeProgramShape({ShapeUtil::MakeShape(F32, {})}, pred_); in TEST_F()358 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()369 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()380 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()563 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()573 ShapeUtil::MakeProgramShape({f32_, f32_}, f32_), {0}); in TEST_F()[all …]
175 compile_result.program_shape = xla::ShapeUtil::MakeProgramShape( in TEST()
270 static ProgramShape MakeProgramShape(std::initializer_list<Shape> parameters,
464 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST()
172 /* static */ ProgramShape ShapeUtil::MakeProgramShape( in MakeProgramShape() function in xla::ShapeUtil