Searched refs:computation_to_apply (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1405 def Call(self, computation_to_apply, operands): argument
1416 return self._client.Call(computation_to_apply.computation, operands)
1443 def Map(self, operands, computation_to_apply, dimensions): argument
1454 return self._client.Map(operands, computation_to_apply.computation,
1457 def Reduce(self, operand, init_value, computation_to_apply, dimensions): argument
1470 computation_to_apply.computation, dimensions)
1472 def ReduceWindow(self, operand, init_value, computation_to_apply, argument
1491 operand, init_value, computation_to_apply.computation,
1495 self, operand, init_value, computation_to_apply, window_dimensions, argument
1513 operand, init_value, computation_to_apply.computation,
|
| D | xla_client_test.py | 1459 computation_to_apply=self._CreateBinaryAddF32Computation(),
1468 computation_to_apply=self._CreateBinaryAddF64Computation(),
1478 computation_to_apply=self._CreateBinaryAddF32Computation(),
1488 computation_to_apply=self._CreateBinaryAddF64Computation(),
1498 computation_to_apply=self._CreateBinaryAddF32Computation(),
1508 computation_to_apply=self._CreateBinaryAddF64Computation(),
1520 computation_to_apply=self._CreateBinaryAddF32Computation(),
1539 computation_to_apply=self._CreateBinaryAddF64Computation(),
1556 computation_to_apply=self._CreateBinaryAddF32Computation(),
1566 computation_to_apply=self._CreateBinaryAddF32Computation(),
[all …]
|