Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/
H A Doperation_semantics.md71 ## AllReduce section in Operation Semantics
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Djitrt_custom_calls.cc1593 struct AllReduce { struct
1601 static AllReduce Handler() { return AllReduce(); } in Handler()
1640 static bool AllReduce(runtime::KernelContext* ctx, void** args, void** attrs) { in AllReduce() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/
H A Dhlo_control_flow_flattening_test.cc313 TEST_F(HloControlFlowFlatteningTest, AllReduce) { in TEST_F() argument
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Dmulti_threaded_pg.py115 class AllReduce: class
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dsharding_propagation_test.cc1775 TEST_P(ParameterizedMetadataTest, AllReduce) { in TEST_P() argument
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dxla_builder.cc2936 XlaOp XlaBuilder::AllReduce(XlaOp operand, const XlaComputation& computation, in AllReduce() function in xla::XlaBuilder
4669 XlaOp AllReduce(const XlaOp operand, const XlaComputation& computation, in AllReduce() function