Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding_util.cc1018 int64 num_update_window_dims = 0; in ScatterOutputShardingFromUpdate() local
1024 dnums.update_window_dims(num_update_window_dims++)); in ScatterOutputShardingFromUpdate()
1042 int64 num_update_window_dims = 0; in ScatterUpdateShardingFromOutput() local
1048 dnums.update_window_dims(num_update_window_dims++)); in ScatterUpdateShardingFromOutput()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc514 int64 num_update_window_dims = 0; in HandleScatter() local
520 dnums.update_window_dims(num_update_window_dims++)); in HandleScatter()