Home
last modified time | relevance | path

Searched defs:sharded_inputs (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/python/tpu/
H A Dtpu_feed.py573 sharded_inputs, argument
818 def generate_enqueue_ops(self, sharded_inputs): argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc3907 absl::InlinedVector<HloInstruction*, 2> sharded_inputs, replicated_inits; in HandleReduceWindow() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc673 std::vector<NodeOut> sharded_inputs; member