Searched refs:init_value_shape_ (Results 1 – 1 of 1) sorted by relevance
88 init_value_shape_ = ShapeUtil::MakeShape(F32, {}); in SelectAndScatterShapeInferenceTest()98 Shape init_value_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()