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.cc265 auto arg_it = fn->arg_begin(); in make_reader() local
266 Argument *arg_in = &*arg_it; in make_reader()
267 ++arg_it; in make_reader()
269 Argument *arg_out = &*arg_it; in make_reader()
270 ++arg_it; in make_reader()
325 auto arg_it = fn->arg_begin(); in make_writer() local
326 Argument *arg_out = &*arg_it; in make_writer()
327 ++arg_it; in make_writer()
329 Argument *arg_len = &*arg_it; in make_writer()
330 ++arg_it; in make_writer()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dstack_ops_decomposition.cc303 auto arg_it = info.stack_var_arg_to_size_arg.find(i); in HandlePartitionedCallOp() local
304 if (arg_it == info.stack_var_arg_to_size_arg.end()) continue; in HandlePartitionedCallOp()
310 assert(arg_it->second == new_operands.size()); in HandlePartitionedCallOp()
Dtensor_list_ops_decomposition.cc430 auto arg_it = info.buffer_arg_to_size_arg.find(i); in HandlePartitionedCallOp() local
431 if (arg_it == info.buffer_arg_to_size_arg.end()) continue; in HandlePartitionedCallOp()
437 assert(arg_it->second == new_operands.size()); in HandlePartitionedCallOp()
/external/bcc/src/cc/frontends/clang/
Db_frontend_action.cc838 for (auto arg_it = D->param_begin(); arg_it != D->param_end(); arg_it++) { in VisitFunctionDecl() local
839 auto *arg = *arg_it; in VisitFunctionDecl()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc524 auto arg_it = kernel->arg_begin(); in BuildKernelPrototype() local
527 llvm::Argument* fn_arg = &*arg_it; in BuildKernelPrototype()
528 ++arg_it; in BuildKernelPrototype()
3138 auto arg_it = kernel->arg_begin(); in BuildKernelThunkImpl() local
3140 for (; arg_it != kernel->arg_end(); ++arg_it, ++buffers_it) { in BuildKernelThunkImpl()
3141 kernel_args[*buffers_it] = arg_it; in BuildKernelThunkImpl()
3159 arg_it->getArgNo(), in BuildKernelThunkImpl()
3160 llvm::Attribute::get(arg_it->getContext(), llvm::Attribute::NoAlias)); in BuildKernelThunkImpl()