Searched defs:AllToAll (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
D | operation_semantics.md | 32 ## AllToAll section in Operation Semantics
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | local_computation_builder.cc | 498 LocalOp ComputationBuilder::AllToAll( in AllToAll() function in xla::swig::ComputationBuilder
|
D | xla_client.py | 1205 def AllToAll(self, member in ComputationBuilder
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder_test.cc | 363 TEST_F(XlaBuilderTest, AllToAll) { in TEST_F() argument
|
D | xla_builder.cc | 2159 XlaOp XlaBuilder::AllToAll(const XlaOp& operand, int64 split_dimension, in AllToAll() function in xla::XlaBuilder 3244 XlaOp AllToAll(const XlaOp& operand, int64 split_dimension, in AllToAll() function
|
/external/tensorflow/tensorflow/go/op/ |
D | wrappers.go | 25654 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp… func
|