| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 80 XlaCompiler::CompilationResult update_computation; in Compile() local
84 &update_computation)); in Compile()
88 ctx->Input("updates"), *update_computation.computation, in Compile()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 435 def scatter(operand, scatter_indices, updates, update_computation, argument
441 update_computation=update_computation,
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 696 xla::XlaComputation update_computation; in ExportXlaOp() local
697 if (failed(ctx.converter->LowerRegionAsComputation(&op.update_computation(), in ExportXlaOp()
698 &update_computation))) { in ExportXlaOp()
705 value_map[op.updates()], update_computation, dimension_numbers, in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1623 update_computation, argument
1629 update_computation.computation, dimension_numbers,
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 588 const XlaComputation& update_computation,
1009 const XlaComputation& update_computation,
1889 const XlaComputation& update_computation,
|
| D | xla_builder.cc | 1803 const XlaComputation& update_computation, in Scatter() argument
1817 update_computation.GetProgramShape()); in Scatter()
1826 AddCalledComputation(update_computation, &instr); in Scatter()
3498 const XlaOp updates, const XlaComputation& update_computation, in Scatter() argument
3502 update_computation, dimension_numbers, in Scatter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1521 auto* update_computation = in TEST_F() local
1527 update_computation, in TEST_F()
|
| D | hlo_parser.cc | 1752 optional<HloComputation*> update_computation; in ParseInstructionRhs() local
1754 &update_computation}; in ParseInstructionRhs()
1776 /*updates=*/operands[2], *update_computation, dim_numbers, in ParseInstructionRhs()
|
| D | hlo_instruction.h | 881 HloComputation* update_computation,
|
| D | hlo_instructions.cc | 2535 HloComputation* update_computation, in HloScatterInstruction() argument
2544 AppendComputation(update_computation); in HloScatterInstruction()
|
| D | hlo_instructions.h | 1497 HloComputation* update_computation,
|
| D | hlo_instruction.cc | 1429 HloComputation* update_computation, in CreateScatter() argument
1433 shape, operand, scatter_indices, updates, update_computation, in CreateScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2278 updated with the values in `updates` using `update_computation`.
2283 <b> `scatter(operand, scatter_indices, updates, update_computation, index_vector_dim, update_window…
2290 `update_computation` | `XlaComputation` | Computation to be used for combining the exi…
2371 `output`[`O`] = `update_computation`(`output`[`O`], `updates`[`U`])
2377 Note that the first parameter that is passed into the `update_computation` will
2380 specifically for cases when the `update_computation` is _not commutative_.
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | hlo_ops.td | 1025 let regions = (region SizedRegion<1>:$update_computation);
|
| D | hlo_ops_base.td | 979 updated with the values in `updates` using `update_computation`.
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 2303 }(&scatter.update_computation()); in matchAndRewrite()
3142 &scatter.update_computation(), &rewriter); in matchAndRewrite()
|