Home
last modified time | relevance | path

Searched refs:map_dims (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_creation_utils.cc364 std::vector<int64_t> map_dims(max_operand_rank); in MakeMapHlo() local
365 std::iota(map_dims.begin(), map_dims.end(), 0); in MakeMapHlo()
369 operand_shapes, map_computation->ComputeProgramShape(), map_dims)); in MakeMapHlo()
Dhlo_verifier.cc1248 std::vector<int64_t> map_dims(max_operand_rank); in HandleMap() local
1249 std::iota(map_dims.begin(), map_dims.end(), 0); in HandleMap()
1254 operand_shapes, map->to_apply()->ComputeProgramShape(), map_dims))); in HandleMap()