Searched defs:update_computation (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | gather_scatter_ops.cc | 81 XlaCompiler::CompilationResult update_computation; in Compile() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.h | 487 const HloComputation* update_computation; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| H A D | xla.py | 582 def scatter(operand, scatter_indices, updates, update_computation, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | xla_builder.cc | 2428 const XlaComputation& update_computation, in Scatter()
2438 const XlaComputation& update_computation, in Scatter()
2476 absl::Span<const XlaOp> updates, const XlaComputation& update_computation, in ScatterInternal()
4907 const XlaOp updates, const XlaComputation& update_computation, in Scatter()
4917 const XlaComputation& update_computation, in Scatter()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| H A D | mlir_hlo_builder.cc | 396 absl::Span<const XlaOp> updates, const XlaComputation& update_computation, in ScatterInternal()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_instruction.cc | 1938 HloComputation* update_computation, in CreateScatter()
1949 HloComputation* update_computation, in CreateScatter()
|
| H A D | hlo_instruction_test.cc | 1678 auto* update_computation = in TEST_F() local
|
| H A D | hlo_parser.cc | 2861 optional<HloComputation*> update_computation; in CreateInstruction() local
|
| H A D | hlo_instructions.cc | 3142 HloComputation* update_computation, in HloScatterInstruction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/ |
| H A D | mlir_hlo_to_hlo.cc | 1473 xla::XlaComputation update_computation; in ExportXlaOp() local
|