Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc109 std::vector<const HloInstruction*> non_io_hlos; in CodegenNestedComputation() local
110 non_io_hlos.push_back(root); in CodegenNestedComputation()
114 non_io_hlos.push_back(hlo); in CodegenNestedComputation()
117 bindings_.EmitBasePointersForHlos(io_hlos, non_io_hlos); in CodegenNestedComputation()
Dhlo_to_ir_bindings.cc40 absl::Span<const HloInstruction* const> non_io_hlos) { in EmitBasePointersForHlos() argument
57 !absl::c_count(non_io_hlos, io_hlo)) in EmitBasePointersForHlos()
70 for (const HloInstruction* non_io_hlo : non_io_hlos) { in EmitBasePointersForHlos()
Dhlo_to_ir_bindings.h48 absl::Span<const HloInstruction* const> non_io_hlos);