Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dhost_callback.cc22 Status HostCallbackContext::OnSend(int arg_num, in OnSend()
110 for (int arg_num = 0; arg_num < hb.operands.size(); ++arg_num) { in CreateHostCallbackStateAndAppendSendRecvCallbacks() local
/external/tensorflow/tensorflow/core/ir/importexport/
Dfunctiondef_export.cc56 int arg_num = block_operand.getArgNumber(); in GetValueName() local
210 for (int arg_num : llvm::seq<int>(0, func_op.getNumArguments())) { in ConvertGenericFunctionToFunctionDef() local
Dfunctiondef_import.cc353 int arg_num = 0; in ImportGenericFunction() local
/external/libbpf/src/
Dusdt.c1282 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
1330 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
1383 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
1477 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
1548 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
1594 static int parse_usdt_arg(const char *arg_str, int arg_num, struct usdt_arg_spec *arg, int *arg_sz) in parse_usdt_arg()
Dusdt.bpf.h116 int bpf_usdt_arg(struct pt_regs *ctx, __u64 arg_num, long *res) in bpf_usdt_arg()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dbuffer_donation_test.cc97 for (int arg_num = 0; arg_num < argument_literals.size(); arg_num++) { in RunAndCheck() local
151 int arg_num = alias.parameter_number; in RunAndCheck() local
/external/bcc/libbpf-tools/android/
Dargp.h44 int arg_num; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_space_to_depth_pass.cc60 unsigned arg_num; member
436 unsigned arg_num) { in HandleCluster()
635 auto arg_num = arg_num_and_num_users.getValue().arg_num; in runOnOperation() local
655 auto arg_num = argnum_and_convolution.getFirst(); in runOnOperation() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_resource.h69 int arg_num() const { return arg_num_; } in arg_num() function
Dxla_resource.cc63 Kind kind, int arg_num, string name, DataType type, TensorShape shape, in XlaResource()
/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/tensorflow/tensorflow/python/platform/
Dbenchmark.py51 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/pytorch/torch/csrc/jit/runtime/
Dvararg_functions.cpp158 auto parse_sublist = [&ss](const c10::List<int64_t>& l, size_t arg_num) { in einsum()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_launch_util.cc258 int arg_num = compilation_result->input_mapping[i]; in PopulateInputs() local
/external/python/cpython3/Modules/
D_posixsubprocess.c1079 for (Py_ssize_t arg_num = 0; arg_num < num_args; ++arg_num) { in subprocess_fork_exec_impl() local
/external/tensorflow/tensorflow/core/ir/
Dops.cc766 for (int arg_num = 0, e = args.size(); arg_num < e; arg_num += 2) { in getAsmBlockArgumentNames() local
/external/ethtool/
Dethtool.c4037 unsigned int arg_num = 0; in do_grxfh() local
4232 u32 arg_num = 0, indir_bytes = 0; in do_srxfh() local