Home
last modified time | relevance | path

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

/external/bcc/src/cc/
Dbpf_module_rw_engine.cc246 auto arg_it = fn->arg_begin(); in make_reader() local
247 Argument *arg_in = &*arg_it; in make_reader()
248 ++arg_it; in make_reader()
250 Argument *arg_out = &*arg_it; in make_reader()
251 ++arg_it; in make_reader()
306 auto arg_it = fn->arg_begin(); in make_writer() local
307 Argument *arg_out = &*arg_it; in make_writer()
308 ++arg_it; in make_writer()
310 Argument *arg_len = &*arg_it; in make_writer()
311 ++arg_it; in make_writer()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc296 auto arg_it = info.stack_var_arg_to_size_arg.find(i); in HandlePartitionedCallOp() local
297 if (arg_it == info.stack_var_arg_to_size_arg.end()) continue; in HandlePartitionedCallOp()
303 assert(arg_it->second == new_operands.size()); in HandlePartitionedCallOp()
Dtensor_list_ops_decomposition.cc446 auto arg_it = info.buffer_arg_to_size_arg.find(i); in HandlePartitionedCallOp() local
447 if (arg_it == info.buffer_arg_to_size_arg.end()) continue; in HandlePartitionedCallOp()
453 assert(arg_it->second == new_operands.size()); in HandlePartitionedCallOp()
/external/bcc/src/cc/frontends/clang/
Db_frontend_action.cc712 for (auto arg_it = D->param_begin(); arg_it != D->param_end(); arg_it++) { in VisitFunctionDecl() local
713 auto *arg = *arg_it; in VisitFunctionDecl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc578 auto arg_it = kernel->arg_begin(); in BuildKernelPrototype() local
581 llvm::Argument* fn_arg = &*arg_it; in BuildKernelPrototype()
582 ++arg_it; in BuildKernelPrototype()
3387 auto arg_it = kernel->arg_begin(); in BuildKernelThunkFromBufferSlices() local
3389 for (; arg_it != kernel->arg_end(); ++arg_it, ++buffers_it) { in BuildKernelThunkFromBufferSlices()
3390 kernel_args[*buffers_it] = arg_it; in BuildKernelThunkFromBufferSlices()
3408 arg_it->getArgNo(), in BuildKernelThunkFromBufferSlices()
3409 llvm::Attribute::get(arg_it->getContext(), llvm::Attribute::NoAlias)); in BuildKernelThunkFromBufferSlices()