Searched refs:get_op_sharding (Results 1 – 1 of 1) sorted by relevance
336 def get_op_sharding(op): function364 return get_op_sharding(tensor.op)