Searched refs:map_dims (Results 1 – 2 of 2) sorted by relevance
364 std::vector<int64_t> map_dims(max_operand_rank); in MakeMapHlo() local365 std::iota(map_dims.begin(), map_dims.end(), 0); in MakeMapHlo()369 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
1248 std::vector<int64_t> map_dims(max_operand_rank); in HandleMap() local1249 std::iota(map_dims.begin(), map_dims.end(), 0); in HandleMap()1254 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()