Home
last modified time | relevance | path

Searched refs:non_io_hlo (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc70 for (const HloInstruction* non_io_hlo : non_io_hlos) { in EmitBasePointersForHlos() local
71 if (already_bound_for_this_function.contains(non_io_hlo)) { in EmitBasePointersForHlos()
74 already_bound_for_this_function.insert(non_io_hlo); in EmitBasePointersForHlos()
76 if (non_io_hlo->opcode() == HloOpcode::kGetTupleElement) { in EmitBasePointersForHlos()
81 non_io_hlo->shape(), in EmitBasePointersForHlos()
83 if (non_io_hlo->opcode() == HloOpcode::kConstant) { in EmitBasePointersForHlos()
85 llvm_ir::ConstantHloToGlobalName(*non_io_hlo)); in EmitBasePointersForHlos()
87 << llvm_ir::ConstantHloToGlobalName(*non_io_hlo); in EmitBasePointersForHlos()
88 BindHloToIrValue(*non_io_hlo, global_for_constant); in EmitBasePointersForHlos()
91 llvm_ir::ShapeToIrType(non_io_hlo->shape(), module_); in EmitBasePointersForHlos()
[all …]