Home
last modified time | relevance | path

Searched defs:ReduceScatter (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/
H A Doperation_semantics.md2146 ## ReduceScatter section in Operation Semantics
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc1790 struct ReduceScatter { struct
1798 static ReduceScatter Handler() { return ReduceScatter(); } in Handler()
1836 static bool ReduceScatter(runtime::KernelContext* ctx, void** args, in ReduceScatter() function
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py191 class ReduceScatter: class
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcollective_ops_test.cc1021 XLA_TEST_F(CollectiveOpsTest, ReduceScatter) { in XLA_TEST_F() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder_test.cc410 TEST_F(XlaBuilderTest, ReduceScatter) { in TEST_F() argument
H A Dxla_builder.cc3017 XlaOp XlaBuilder::ReduceScatter( in ReduceScatter() function in xla::XlaBuilder
4679 XlaOp ReduceScatter(const XlaOp operand, const XlaComputation& computation, in ReduceScatter() function