Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc286 int64 current_after_dim = in AddInstruction() local
291 reshape_broadcast_dims.erase(current_after_dim); in AddInstruction()
304 for (int64 i = current_after_dim; i < hlo->shape().rank(); ++i) { in AddInstruction()