Home
last modified time | relevance | path

Searched defs:AllGather (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/
H A Doperation_semantics.md32 ## AllGather section in Operation Semantics
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc1858 struct AllGather { struct
1866 static AllGather Handler() { return AllGather(); } in Handler()
1902 static bool AllGather(runtime::KernelContext* ctx, void** args, void** attrs) { in AllGather() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_control_flow_flattening_test.cc372 TEST_F(HloControlFlowFlatteningTest, AllGather) { in TEST_F() argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py142 class AllGather: class
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2868 XlaOp XlaBuilder::AllGather(XlaOp operand, int64_t all_gather_dimension, in AllGather() function in xla::XlaBuilder
4653 XlaOp AllGather(const XlaOp operand, int64_t all_gather_dimension, in AllGather() function