spmd_full_to_shard_shape (reference) in projects: external

Searched refs:spmd_full_to_shard_shape (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
D	xla_sharding.py	`469 return tf2xla.spmd_full_to_shard_shape(`
/external/tensorflow/tensorflow/compiler/tf2xla/python/
D	xla.py	`533 spmd_full_to_shard_shape = gen_xla_ops.xla_spmd_full_to_shard_shape variable`