Searched refs:scatter_program_shape_ (Results 1 – 1 of 1) sorted by relevance
95 scatter_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()104 ProgramShape scatter_program_shape_; member in xla::__anon5c819e1b0111::SelectAndScatterShapeInferenceTest344 init_value_shape_, scatter_program_shape_); in TEST_F()354 init_value_shape_, scatter_program_shape_); in TEST_F()365 init_value_shape_, scatter_program_shape_); in TEST_F()376 init_value_shape_, scatter_program_shape_); in TEST_F()387 init_value_shape_, scatter_program_shape_); in TEST_F()398 init_value_shape_, scatter_program_shape_); in TEST_F()