Searched defs:AllToAll (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/security/advisory/ |
| H A D | tfsa-2021-176.md | 3 ### CVE Number
6 ### Impact
36 ### Patches
41 ### For more information
44 ### Attribution
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | tpu_cross_replica_ops_test.cc | 24 TEST(AllToAll, UnknownRank) { in TEST() argument
31 TEST(AllToAll, KnownRankUnknownDims) { in TEST() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | all_to_all.cc | 42 AllToAll::AllToAll() in AllToAll() function in tensorflow::AllToAll
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| H A D | operation_semantics.md | 115 ## AllToAll section in Operation Semantics
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | jitrt_custom_calls.cc | 1922 struct AllToAll { struct
1930 static AllToAll Handler() { return AllToAll(); } in Handler()
1967 static bool AllToAll(runtime::KernelContext* ctx, void** args, void** attrs) { in AllToAll() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | hlo_control_flow_flattening_test.cc | 395 TEST_F(HloControlFlowFlatteningTest, AllToAll) { in TEST_F() argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | multi_threaded_pg.py | 69 class AllToAll: class
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder_test.cc | 436 TEST_F(XlaBuilderTest, AllToAll) { in TEST_F() argument
|
| H A D | xla_builder.cc | 3077 XlaOp XlaBuilder::AllToAll(XlaOp operand, int64_t split_dimension, in AllToAll() function in xla::XlaBuilder
4690 XlaOp AllToAll(const XlaOp operand, int64_t split_dimension, in AllToAll() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/go/op/ |
| H A D | wrappers.go | 662 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp… func
|