Searched refs:MakeProgramShape (Results 1 – 5 of 5) sorted by relevance
66 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_, f32_}, f32_); in ExpectInferredReduceShape()91 select_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()93 scatter_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()329 ProgramShape to_apply = ShapeUtil::MakeProgramShape( in TEST_F()360 ShapeUtil::MakeProgramShape({ShapeUtil::MakeShape(F32, {})}, pred_); in TEST_F()370 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()381 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()392 ProgramShape select_program_shape_fail = ShapeUtil::MakeProgramShape( in TEST_F()944 ProgramShape to_apply = ShapeUtil::MakeProgramShape({f32_}, s32_); in TEST_F()954 ShapeUtil::MakeProgramShape({f32_, f32_}, f32_), {0}); in TEST_F()[all …]
183 ProgramShape prog = ShapeUtil::MakeProgramShape( in TEST_F()
350 static ProgramShape MakeProgramShape(std::initializer_list<Shape> parameters,
242 /* static */ ProgramShape ShapeUtil::MakeProgramShape( in MakeProgramShape() function in xla::ShapeUtil
231 xla::ShapeUtil::MakeProgramShape( in TEST()