Searched defs:from_dev_matrix (Results 1 – 2 of 2) sorted by relevance
336 def conver_tensor_by_layout(from_dev_matrix, from_tensor_map, from_opt_shard_step, from_opt_shard_s… argument
359 def _construct_from_to_tensor_layout(from_full_tensor_shape, from_dev_matrix, argument