Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md32 ## AllToAll section in Operation Semantics
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc498 LocalOp ComputationBuilder::AllToAll( in AllToAll() function in xla::swig::ComputationBuilder
Dxla_client.py1205 def AllToAll(self, member in ComputationBuilder
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder_test.cc363 TEST_F(XlaBuilderTest, AllToAll) { in TEST_F() argument
Dxla_builder.cc2159 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/
Dwrappers.go25654 func AllToAll(scope *Scope, input tf.Output, group_assignment tf.Output, concat_dimension int64, sp… func