Searched refs:scatter_program_shape_ (Results 1 – 1 of 1) sorted by relevance
91 scatter_program_shape_ = ShapeUtil::MakeProgramShape( in SelectAndScatterShapeInferenceTest()100 ProgramShape scatter_program_shape_; member in xla::__anon8d0540a50111::SelectAndScatterShapeInferenceTest330 init_value_shape_, scatter_program_shape_); in TEST_F()340 init_value_shape_, scatter_program_shape_); in TEST_F()351 init_value_shape_, scatter_program_shape_); in TEST_F()362 init_value_shape_, scatter_program_shape_); in TEST_F()373 init_value_shape_, scatter_program_shape_); in TEST_F()384 init_value_shape_, scatter_program_shape_); in TEST_F()