Searched defs:update_computation (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 81 XlaCompiler::CompilationResult update_computation; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.h | 414 const HloComputation* update_computation; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 561 def scatter(operand, scatter_indices, updates, update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 319 const XlaComputation& update_computation, in ScatterInternal()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1061 xla::XlaComputation update_computation; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1579 auto* update_computation = in TEST_F() local
|
| D | hlo_parser.cc | 2656 optional<HloComputation*> update_computation; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1814 HloComputation* update_computation, in CreateScatter()
|
| D | hlo_instructions.cc | 2983 HloComputation* update_computation, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 2331 const XlaComputation& update_computation, in Scatter()
2353 const XlaComputation& update_computation, in ScatterInternal()
4745 const XlaOp updates, const XlaComputation& update_computation, in Scatter()
|