Searched refs:MakeProgramShape (Results 1 – 5 of 5) sorted by relevance
65 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()90 select_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()92 scatter_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()328 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F()359 ShapeUtil::MakeProgramShape({ShapeUtil::MakeShape(F32, {})}, pred_); 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()391 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()757 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()767 ShapeUtil::MakeProgramShape({f32_, f32_}, f32_), {0}); in TEST_F()[all …]
209 xla::ShapeUtil::MakeProgramShape( in TEST()
182 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST_F()
307 static ProgramShape MakeProgramShape(std::initializer_list<Shape> parameters,
164 /* static */ ProgramShape ShapeUtil::MakeProgramShape( in MakeProgramShape() function in xla::ShapeUtil