Searched defs:from_tensor_map (Results 1 – 3 of 3) sorted by relevance
336 def conver_tensor_by_layout(from_dev_matrix, from_tensor_map, from_opt_shard_step, from_opt_shard_s… argument
360 from_tensor_map, to_full_tensor_shape, argument
335 auto from_tensor_map = from_layout.tensor_map(); in SpecialPatternInTransformer() local