Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1015 std::vector<int64> cw_start_indices(dot->shape().rank(), 0); in PartitionBaseCase() local
1016 cw_start_indices[slice_dim] = dot->shape().dimensions(slice_dim) / 2; in PartitionBaseCase()
1021 slice_shape, dot, cw_start_indices, dot->shape().dimensions(), in PartitionBaseCase()