Searched defs:tensor_split_dims_mapping (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/ | ||
| H A D | xla_sharding.py | 508 tensor_split_dims_mapping, argument |