HomeSort by relevance Sort by last modified time
    Searched defs:CrossReplicaSum (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/python/
local_computation_builder.cc 358 ComputationDataHandle LocalComputationBuilder::CrossReplicaSum(
360 return builder_.CrossReplicaSum(operand);
xla_client.py 674 def CrossReplicaSum(self, operand):
675 """CrossReplicaSum op.
684 self._client.CrossReplicaSum(_unwrap_data_handle(operand)))
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
computation_builder.cc     [all...]

Completed in 235 milliseconds