Searched refs:map_dims (Results 1 – 2 of 2) sorted by relevance
240 std::vector<int64> map_dims(max_operand_rank); in MakeMapHlo() local241 std::iota(map_dims.begin(), map_dims.end(), 0); in MakeMapHlo()245 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
608 std::vector<int64> map_dims(max_operand_rank); in HandleMap() local609 std::iota(map_dims.begin(), map_dims.end(), 0); in HandleMap()614 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()