Home
last modified time | relevance | path

Searched defs:arg_spec (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dfunction_api_info_test.cc41 for (const auto& arg_spec : input_args_spec) in SetArgs() local
43 for (const auto& arg_spec : output_args_spec) in SetArgs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dop_converter.h142 const InputArgSpec arg_spec = Impl::InputSpec()[i]; in ValidateInputs() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dcompiler.cpp202 const ArgSpec& arg_spec, in compileKernel()
H A Dexecutor.cpp378 ArgSpec arg_spec{inputs, device.index()}; in runFusion() local
/aosp_15_r20/external/libbpf/src/
H A Dusdt.bpf.h119 struct __bpf_usdt_arg_spec *arg_spec; in bpf_usdt_arg() local
/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Ddeprecation.py443 def _get_deprecated_positional_arguments(names_to_ok_vals, arg_spec): argument
/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dvmap.py392 def _concat_chunked_outputs(out_dims, arg_spec, flat_output_chunks): argument