Home
last modified time | relevance | path

Searched defs:reshard_operand (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2408 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleSlice() local
2729 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleReshape() local
3413 auto reshard_operand = in HandlePad() local
H A Dspmd_partitioner_test.cc1427 auto reshard_operand = op::Reshape(op::Transpose( in TEST_F() local