Searched defs:update_computation (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 764 const LocalOp& updates, const Computation& update_computation, in Scatter()
|
| D | xla_client.py | 1779 def Scatter(self, a, scatter_indices, updates, update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1448 auto* update_computation = in TEST_F() local
|
| D | hlo_parser.cc | 1663 optional<HloComputation*> update_computation; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1317 HloComputation* update_computation, in CreateScatter()
|
| D | hlo_instructions.cc | 2397 HloComputation* update_computation, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1866 const XlaComputation& update_computation, in Scatter()
3432 const XlaOp& updates, const XlaComputation& update_computation, in Scatter()
|