Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dir_function.cc147 llvm::Function::arg_iterator arg_iter = function_->arg_begin(); in Initialize() local
148 arg_iter->setName("retval"); in Initialize()
149 result_arg_ = &*arg_iter; in Initialize()
150 (++arg_iter)->setName("run_options"); in Initialize()
151 exec_run_options_arg_ = &*arg_iter; in Initialize()
152 (++arg_iter)->setName("params"); in Initialize()
153 parameters_arg_ = &*arg_iter; in Initialize()
154 (++arg_iter)->setName("buffer_table"); in Initialize()
155 buffer_table_arg_ = &*arg_iter; in Initialize()
157 (++arg_iter)->setName("dynamic_loop_bounds"); in Initialize()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhlo_to_ir_bindings.cc54 auto arg_iter = function->arg_begin(); in EmitBasePointersForHlos() local
60 BindHloToIrValue(*io_hlo, &*arg_iter); in EmitBasePointersForHlos()
63 ++arg_iter; in EmitBasePointersForHlos()
67 arg_iter->setName("output_arg"); in EmitBasePointersForHlos()
68 ++arg_iter; in EmitBasePointersForHlos()