Searched defs:update_computation (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | gather_scatter_ops.cc | 80 XlaCompiler::CompilationResult update_computation; in Compile() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/python/ |
| D | xla.py | 435 def scatter(operand, scatter_indices, updates, update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 696 xla::XlaComputation update_computation; in ExportXlaOp() local
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 1623 update_computation, argument
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_instruction_test.cc | 1521 auto* update_computation = in TEST_F() local
|
| D | hlo_parser.cc | 1752 optional<HloComputation*> update_computation; in ParseInstructionRhs() local
|
| D | hlo_instruction.cc | 1429 HloComputation* update_computation, in CreateScatter()
|
| D | hlo_instructions.cc | 2535 HloComputation* update_computation, in HloScatterInstruction()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.cc | 1803 const XlaComputation& update_computation, in Scatter()
3498 const XlaOp updates, const XlaComputation& update_computation, in Scatter()
|