Searched refs:map_dims (Results 1 – 2 of 2) sorted by relevance
312 std::vector<int64> map_dims(max_operand_rank); in MakeMapHlo() local313 std::iota(map_dims.begin(), map_dims.end(), 0); in MakeMapHlo()317 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
878 std::vector<int64> map_dims(max_operand_rank); in HandleMap() local879 std::iota(map_dims.begin(), map_dims.end(), 0); in HandleMap()884 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()