Searched refs:MakeProgramShape (Results 1 – 5 of 5) sorted by relevance
64 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()89 select_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()91 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()567 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()577 ShapeUtil::MakeProgramShape({f32_, f32_}, f32_), {0}); in TEST_F()[all …]
180 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST_F()
305 static ProgramShape MakeProgramShape(std::initializer_list<Shape> parameters,
148 /* static */ ProgramShape ShapeUtil::MakeProgramShape( in MakeProgramShape() function in xla::ShapeUtil
196 xla::ShapeUtil::MakeProgramShape( in TEST()