Searched defs:indices_strategy (Results 1 – 3 of 3) sorted by relevance
143 Dimensions indices_strategy = sp->GetInputDim()[0]; in GenerateOpStrategies() local
163 Dimensions indices_strategy(inputs_shape_[1].size(), 1); in GenerateOpStrategies() local
179 Dimensions indices_strategy = strategy[1]; in CheckManualSplit() local268 Dimensions indices_strategy = strategy[1]; in ShardBatchAndAxis() local