Home
last modified time | relevance | path

Searched defs:arg_num (Results 1 – 18 of 18) sorted by relevance

/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()
108 EXTERN void __tgt_target_data_begin_nowait(int64_t device_id, int32_t arg_num, in __tgt_target_data_begin_nowait()
120 int32_t arg_num, void **args_base, 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()
173 EXTERN void __tgt_target_data_end(int64_t device_id, int32_t arg_num, in __tgt_target_data_end()
179 EXTERN void __tgt_target_data_end_nowait(int64_t device_id, int32_t arg_num, in __tgt_target_data_end_nowait()
191 int32_t arg_num, void **args_base, in __tgt_target_data_end_mapper()
235 ident_t *loc, int64_t device_id, int32_t arg_num, void **args_base, in __tgt_target_data_end_nowait_mapper()
246 EXTERN void __tgt_target_data_update(int64_t device_id, int32_t arg_num, in __tgt_target_data_update()
252 EXTERN void __tgt_target_data_update_nowait(int64_t device_id, int32_t arg_num, in __tgt_target_data_update_nowait()
[all …]
Domptarget.cpp250 int targetDataBegin(DeviceTy &Device, int32_t arg_num, void **args_base, in targetDataBegin()
/external/llvm-project/openmp/libomptarget/test/unified_shared_memory/
Dclose_manual.c51 int arg_num = 1; in main() local
/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()
317 int32_t arg_num) { in __tgt_rtl_run_target_region()
/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
148 int arg_num = alias.parameter_number; in RunAndCheck() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h66 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()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc58 unsigned arg_num; member
481 unsigned arg_num) { in HandleCluster()
680 auto arg_num = arg_num_and_num_users.getValue().arg_num; in runOnOperation() local
700 auto arg_num = argnum_and_convolution.getFirst(); in runOnOperation() local
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dfusion_utils.cc224 auto arg_num = std::distance(second_inputs.begin(), input_it); in FuseFunctionNodes() local
316 const StringCollection& first_outputs, int arg_num) { in ComposeInput()
357 const StringCollection& first_outputs, int arg_num) { in SameInput()
Dfusion_utils_test.cc167 const StringCollection &parent_outputs, int arg_num) { in TEST()
/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()
1131 ptrdiff_t *tgt_offsets, int32_t arg_num, int32_t team_num, in __tgt_rtl_run_target_team_region_async()
1143 int32_t arg_num) { in __tgt_rtl_run_target_region()
1158 int32_t arg_num, in __tgt_rtl_run_target_region_async()
/external/llvm-project/openmp/libomptarget/plugins/amdgpu/src/
Drtl.cpp1200 uint32_t arg_num = kernarg_segment_size / sizeof(void *); in __tgt_rtl_load_binary_locked() local
1580 int32_t arg_num, int32_t 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()
1768 int32_t arg_num) { in __tgt_rtl_run_target_region()
1781 int32_t arg_num, in __tgt_rtl_run_target_region_async()
/external/tensorflow/tensorflow/python/platform/
Dbenchmark.py56 def _rename_function(f, arg_num, name): argument
/external/tensorflow/tensorflow/core/common_runtime/
Dreplicate_per_replica_nodes_test.cc59 int arg_num = 0; in CheckArgNum() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc233 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs() local
/external/python/cpython3/Modules/
D_posixsubprocess.c634 Py_ssize_t arg_num, num_groups = 0; in subprocess_fork_exec() local
/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
/external/ethtool/
Dethtool.c3829 int arg_num = 0; in do_grxfh() local
4028 u32 arg_num = 0, indir_bytes = 0; in do_srxfh() local