Searched refs:update_computation (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 764 const LocalOp& updates, const Computation& update_computation, in Scatter() argument
767 update_computation.computation(), dimension_numbers); in Scatter()
|
| D | local_computation_builder.h | 378 const LocalOp& updates, const Computation& update_computation,
|
| D | xla_client.py | 1779 def Scatter(self, a, scatter_indices, updates, update_computation, argument
1783 a, scatter_indices, updates, update_computation.computation,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 548 const XlaOp& updates, const XlaComputation& update_computation,
973 const XlaComputation& update_computation,
1821 const XlaOp& updates, const XlaComputation& update_computation,
|
| D | xla_builder.cc | 1866 const XlaComputation& update_computation, in Scatter() argument
1876 update_computation.GetProgramShape()); in Scatter()
1885 AddCalledComputation(update_computation, &instr); in Scatter()
3432 const XlaOp& updates, const XlaComputation& update_computation, in Scatter() argument
3435 update_computation, dimension_numbers); in Scatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1448 auto* update_computation = in TEST_F() local
1454 update_computation, in TEST_F()
|
| D | hlo_instruction.h | 734 HloComputation* update_computation,
|
| D | hlo_parser.cc | 1663 optional<HloComputation*> update_computation; in ParseInstructionRhs() local
1665 &update_computation}; in ParseInstructionRhs()
1681 /*updates=*/operands[2], *update_computation, dim_numbers)); in ParseInstructionRhs()
|
| D | hlo_instructions.cc | 2397 HloComputation* update_computation, in HloScatterInstruction() argument
2403 AppendComputation(update_computation); in HloScatterInstruction()
|
| D | hlo_instructions.h | 1404 HloComputation* update_computation,
|
| D | hlo_instruction.cc | 1317 HloComputation* update_computation, in CreateScatter() argument
1320 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2110 updated with the values in `updates` using `update_computation`.
2115 <b> `scatter(operand, scatter_indices, updates, update_computation, index_vector_dim, update_window…
2122 `update_computation` | `XlaComputation` | Computation to be used for combining the exi…
2199 `output`[`O`] = `update_computation`(`output`[`O`], `updates`[`U`])
2205 Note that the first parameter that is passed into the `update_computation` will
2208 specifically for cases when the `update_computation` is _not commutative_.
|