Searched refs:reshard_operand (Results 1 – 2 of 2) sorted by relevance
1601 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleSlice() local1605 if (!reshard_operand.has_value()) { in HandleSlice()1608 TF_RET_CHECK(!reshard_operand->dynamic_slice_index_on_output.has_value()); in HandleSlice()1609 const Shape& operand_shape = reshard_operand->sharded_input->shape(); in HandleSlice()1616 auto dim = reshard_operand->shard_window.dimensions(i); in HandleSlice()1629 shard_shape, reshard_operand->sharded_input, start_indices, in HandleSlice()1632 auto data = reshard_operand->sharded_input; in HandleSlice()2071 auto reshard_operand = operand.ReshardAsWindowedInput( in HandleReshape() local2075 if (!reshard_operand.has_value()) { in HandleReshape()2078 TF_RET_CHECK(!reshard_operand->dynamic_slice_index_on_output.has_value()); in HandleReshape()[all …]
1390 auto reshard_operand = op::Reshape(op::Transpose( in TEST_F() local1398 reshard_operand, op::Broadcast(op::Constant()))); in TEST_F()