Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_nested.cc57 std::vector<const HloInstruction*> io_hlos; in CodegenNestedComputation() local
62 io_hlos.push_back(param); in CodegenNestedComputation()
117 bindings_.EmitBasePointersForHlos(io_hlos, non_io_hlos); in CodegenNestedComputation()
Dhlo_to_ir_bindings.cc39 absl::Span<const HloInstruction* const> io_hlos, in EmitBasePointersForHlos() argument
49 CHECK_EQ(io_hlos.size() + 1, function->arg_size()); in EmitBasePointersForHlos()
55 for (const HloInstruction* io_hlo : io_hlos) { in EmitBasePointersForHlos()
Dhlo_to_ir_bindings.h47 absl::Span<const HloInstruction* const> io_hlos,