Home
last modified time | relevance | path

Searched refs:arg_num (Results 1 – 25 of 26) sorted by relevance

12

/external/llvm-project/openmp/libomptarget/src/
Dinterface.cpp102 EXTERN void __tgt_target_data_begin(int64_t device_id, int32_t arg_num, in __tgt_target_data_begin() argument
104 __tgt_target_data_begin_mapper(nullptr, device_id, arg_num, args_base, args, in __tgt_target_data_begin()
108 EXTERN void __tgt_target_data_begin_nowait(int64_t device_id, int32_t arg_num, in __tgt_target_data_begin_nowait() argument
115 __tgt_target_data_begin_mapper(nullptr, device_id, arg_num, args_base, args, in __tgt_target_data_begin_nowait()
120 int32_t arg_num, void **args_base, in __tgt_target_data_begin_mapper() argument
128 device_id, arg_num); in __tgt_target_data_begin_mapper()
145 for (int i = 0; i < arg_num; ++i) { in __tgt_target_data_begin_mapper()
153 int rc = targetDataBegin(Device, arg_num, args_base, args, arg_sizes, in __tgt_target_data_begin_mapper()
159 ident_t *loc, int64_t device_id, int32_t arg_num, void **args_base, in __tgt_target_data_begin_nowait_mapper() argument
166 __tgt_target_data_begin_mapper(loc, device_id, arg_num, args_base, args, in __tgt_target_data_begin_nowait_mapper()
[all …]
Dprivate.h21 extern int targetDataBegin(DeviceTy &Device, int32_t arg_num, void **args_base,
31 extern int targetDataUpdate(DeviceTy &Device, int32_t arg_num, void **args_base,
Domptarget.cpp250 int targetDataBegin(DeviceTy &Device, int32_t arg_num, void **args_base, in targetDataBegin() argument
255 for (int32_t i = 0; i < arg_num; ++i) { in targetDataBegin()
290 if (getParentIndex(arg_types[i]) < 0 && next_i < arg_num && in targetDataBegin()
/external/llvm-project/openmp/libomptarget/include/
Domptarget.h166 void __tgt_target_data_begin(int64_t device_id, int32_t arg_num,
169 void __tgt_target_data_begin_nowait(int64_t device_id, int32_t arg_num,
176 int32_t arg_num, void **args_base,
182 ident_t *loc, int64_t device_id, int32_t arg_num, void **args_base,
190 void __tgt_target_data_end(int64_t device_id, int32_t arg_num, void **args_base,
192 void __tgt_target_data_end_nowait(int64_t device_id, int32_t arg_num,
198 int32_t arg_num, void **args_base,
203 ident_t *loc, int64_t device_id, int32_t arg_num, void **args_base,
209 void __tgt_target_data_update(int64_t device_id, int32_t arg_num,
212 void __tgt_target_data_update_nowait(int64_t device_id, int32_t arg_num,
[all …]
/external/llvm-project/openmp/libomptarget/test/unified_shared_memory/
Dclose_manual.c14 extern void __tgt_target_data_begin(int64_t device_id, int32_t arg_num,
18 extern void __tgt_target_data_end(int64_t device_id, int32_t arg_num,
51 int arg_num = 1; in main() local
54 int64_t arg_sizes[arg_num]; in main()
58 int64_t arg_types[arg_num]; in main()
67 __tgt_target_data_begin(device_id, arg_num, args_base, args, arg_sizes, in main()
73 __tgt_target_data_end(device_id, arg_num, args_base, args, arg_sizes, in main()
/external/llvm-project/openmp/libomptarget/plugins/generic-elf-64bit/src/
Drtl.cpp281 int32_t arg_num, int32_t team_num, in __tgt_rtl_run_target_team_region() argument
290 std::vector<ffi_type *> args_types(arg_num, &ffi_type_pointer); in __tgt_rtl_run_target_team_region()
291 std::vector<void *> args(arg_num); in __tgt_rtl_run_target_team_region()
292 std::vector<void *> ptrs(arg_num); in __tgt_rtl_run_target_team_region()
294 for (int32_t i = 0; i < arg_num; ++i) { in __tgt_rtl_run_target_team_region()
299 ffi_status status = ffi_prep_cif(&cif, FFI_DEFAULT_ABI, arg_num, in __tgt_rtl_run_target_team_region()
317 int32_t arg_num) { in __tgt_rtl_run_target_region() argument
320 tgt_offsets, arg_num, 1, 1, 0); in __tgt_rtl_run_target_region()
/external/python/asn1crypto/dev/
D_task.py105 arg_num = 1
113 val = _get_arg(arg_num)
125 arg_num += 1
139 max_args = arg_num + 1 + len(run_args)
145 val = _get_arg(arg_num + 1 + i)
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc94 for (int arg_num = 0; arg_num < argument_literals.size(); arg_num++) { in RunAndCheck() local
95 const bool donate_argument = donate_arguments[arg_num]; in RunAndCheck()
96 const Literal& argument_literal = argument_literals[arg_num]; in RunAndCheck()
148 int arg_num = alias.parameter_number; in RunAndCheck() local
150 inputs_buffers[arg_num].element(alias.parameter_index).opaque(); in RunAndCheck()
154 expected_runtime_aliasing[arg_num]); in RunAndCheck()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc58 unsigned arg_num; member
481 unsigned arg_num) { in HandleCluster() argument
488 if (block_arg.getArgNumber() != arg_num) continue; in HandleCluster()
500 if (input.index() != arg_num) continue; in HandleCluster()
680 auto arg_num = arg_num_and_num_users.getValue().arg_num; in runOnOperation() local
682 argnum_and_convolutions[arg_num].emplace_back(conv2d, block_size); in runOnOperation()
683 argnum_num_users[arg_num] = num_users; in runOnOperation()
700 auto arg_num = argnum_and_convolution.getFirst(); in runOnOperation() local
705 if (argnum_num_users[arg_num] != conv2d_and_block_sizes.size() || in runOnOperation()
707 argnum_and_convolutions.erase(arg_num); in runOnOperation()
[all …]
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.h49 const StringCollection& parent_outputs, int arg_num)>;
74 const StringCollection& first_outputs, int arg_num);
106 const StringCollection& first_outputs, int arg_num);
Dfusion_utils.cc224 auto arg_num = std::distance(second_inputs.begin(), input_it); in FuseFunctionNodes() local
226 set_input(first_inputs, second_inputs, first_outputs, arg_num); in FuseFunctionNodes()
316 const StringCollection& first_outputs, int arg_num) { in ComposeInput() argument
318 return first_outputs.at(arg_num); in ComposeInput()
357 const StringCollection& first_outputs, int arg_num) { in SameInput() argument
358 return first_inputs.at(arg_num); in SameInput()
Dfusion_utils_test.cc167 const StringCollection &parent_outputs, int arg_num) { in TEST() argument
169 return function_inputs.at(arg_num); in TEST()
/external/llvm-project/openmp/libomptarget/plugins/amdgpu/src/
Drtl.cpp162 void *allocate(uint64_t arg_num) { in allocate()
163 assert((arg_num * sizeof(void *)) == kernarg_segment_size); in allocate()
1200 uint32_t arg_num = kernarg_segment_size / sizeof(void *); in __tgt_rtl_load_binary_locked() local
1201 std::vector<size_t> arg_sizes(arg_num); in __tgt_rtl_load_binary_locked()
1574 ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t num_teams,
1580 int32_t arg_num, int32_t num_teams, in __tgt_rtl_run_target_team_region() argument
1586 device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, num_teams, in __tgt_rtl_run_target_team_region()
1595 ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t num_teams, in __tgt_rtl_run_target_team_region_locked() argument
1604 std::vector<void *> args(arg_num); in __tgt_rtl_run_target_team_region_locked()
1605 std::vector<void *> ptrs(arg_num); in __tgt_rtl_run_target_team_region_locked()
[all …]
/external/ethtool/
Dethtool.c3829 int arg_num = 0; in do_grxfh() local
3833 while (arg_num < ctx->argc) { in do_grxfh()
3834 if (!strcmp(ctx->argp[arg_num], "context")) { in do_grxfh()
3835 ++arg_num; in do_grxfh()
3836 rss_context = get_int_range(ctx->argp[arg_num], 0, 1, in do_grxfh()
3838 ++arg_num; in do_grxfh()
4028 u32 arg_num = 0, indir_bytes = 0; in do_srxfh() local
4038 while (arg_num < ctx->argc) { in do_srxfh()
4039 if (!strcmp(ctx->argp[arg_num], "equal")) { in do_srxfh()
4040 ++arg_num; in do_srxfh()
[all …]
/external/python/cpython3/Modules/
D_posixsubprocess.c634 Py_ssize_t arg_num, num_groups = 0; in subprocess_fork_exec() local
720 for (arg_num = 0; arg_num < num_args; ++arg_num) { in subprocess_fork_exec()
726 borrowed_arg = PySequence_Fast_GET_ITEM(fast_args, arg_num); in subprocess_fork_exec()
729 PyTuple_SET_ITEM(converted_args, arg_num, converted_arg); in subprocess_fork_exec()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h50 XlaResource(Kind kind, int arg_num, string name, DataType type,
66 int arg_num() const { return arg_num_; } in arg_num() function
Dxla_resource.cc62 XlaResource::XlaResource(Kind kind, int arg_num, string name, DataType type, in XlaResource() argument
68 arg_num_(arg_num), in XlaResource()
Dxla_compiler.cc253 output.input_index = retval.resource()->arg_num(); in BuildComputation()
269 if (resource->arg_num() >= 0) { in BuildComputation()
275 return a->arg_num() < b->arg_num(); in BuildComputation()
285 DCHECK_LT(resource->arg_num(), args.size()); in BuildComputation()
286 const XlaCompiler::Argument& arg = args[resource->arg_num()]; in BuildComputation()
287 auto it = arg_shardings.find(resource->arg_num()); in BuildComputation()
301 update.input_index = resource->arg_num(); in BuildComputation()
/external/tensorflow/tensorflow/python/platform/
Dbenchmark.py56 def _rename_function(f, arg_num, name): argument
60 new_code = types.CodeType(arg_num, func_code.co_nlocals,
70 new_code = types.CodeType(arg_num, func_code.co_posonlyargcount,
79 new_code = types.CodeType(arg_num, 0, func_code.co_nlocals,
/external/llvm-project/openmp/libomptarget/plugins/cuda/src/
Drtl.cpp1114 int32_t arg_num, int32_t team_num, in __tgt_rtl_run_target_team_region() argument
1121 device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num, in __tgt_rtl_run_target_team_region()
1131 ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t team_num, in __tgt_rtl_run_target_team_region_async() argument
1137 device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, team_num, in __tgt_rtl_run_target_team_region_async()
1143 int32_t arg_num) { in __tgt_rtl_run_target_region() argument
1148 device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, &async_info); in __tgt_rtl_run_target_region()
1158 int32_t arg_num, in __tgt_rtl_run_target_region_async() argument
1163 device_id, tgt_entry_ptr, tgt_args, tgt_offsets, arg_num, in __tgt_rtl_run_target_region_async()
/external/llvm-project/openmp/libomptarget/test/offloading/
Dnon_contiguous_update.cpp27 void __tgt_target_data_update(int64_t device_id, int32_t arg_num,
/external/tensorflow/tensorflow/core/common_runtime/
Dreplicate_per_replica_nodes_test.cc59 int arg_num = 0; in CheckArgNum() local
62 arg_num++; in CheckArgNum()
65 EXPECT_EQ(arg_num, expected_num); in CheckArgNum()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc233 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs() local
234 CHECK_GE(arg_num, missing_ctx_input_prefix); in PopulateInputs()
239 bool is_resource_variable = resource_vars.count(arg_num); in PopulateInputs()
248 ? resource_vars.at(arg_num) in PopulateInputs()
249 : &(ctx->input(arg_num - missing_ctx_input_prefix)); in PopulateInputs()
/external/bcc/src/cc/frontends/b/
Dcodegen_llvm.cc259 auto arg_num = stoi(n->sub_name_.substr(3, 3)); in visit_ident_expr_node() local
260 if (arg_num < 5) { in visit_ident_expr_node()
261 emit("%s%s->args_lo[%d]", n->decl_->scope_id(), n->c_str(), arg_num - 1); in visit_ident_expr_node()
263 emit("%s%s->args_hi[%d]", n->decl_->scope_id(), n->c_str(), arg_num - 5); in visit_ident_expr_node()
/external/llvm/test/Transforms/LICM/
Dsinking.ll208 …%arg_num.0.i.ph13000 = phi i32 [ 0, %loopentry.2.i ], [ %inc.1.i, %loopentry.3.i.preheader.loopexi…

12