spmd_shard_to_full_shape (reference) in projects: external

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

/external/tensorflow/tensorflow/compiler/xla/experimental/xla_sharding/
D	xla_sharding.py	`405 return tf2xla.spmd_shard_to_full_shape(`
/external/tensorflow/tensorflow/compiler/tf2xla/python/
D	xla.py	`456 spmd_shard_to_full_shape = gen_xla_ops.xla_spmd_shard_to_full_shape variable`