Home
last modified time | relevance | path

Searched defs:num_arguments (Results 1 – 25 of 26) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc241 size_t& num_arguments, size_t& num_results) { in CreateOperands()
265 const HloInstruction* instr, size_t& num_arguments, size_t& num_results, in CreateOpWithoutAttrs()
276 size_t num_arguments, size_t num_results) { in CreateOpInFusion()
329 size_t num_arguments, num_results; in CreateOpInFusion() local
718 size_t num_arguments, num_results; in EmitCustomCallOp() local
1382 size_t num_arguments, num_results; in EmitCaseOp() local
/aosp_15_r20/external/pytorch/aten/src/ATen/functorch/
H A DBatchedFallback.cpp102 const auto num_arguments = schema.arguments().size(); in batchedTensorInplaceForLoopFallback() local
262 const auto num_arguments = schema.arguments().size(); in batchedTensorForLoopFallback() local
402 const auto num_arguments = schema.arguments().size(); in batchedNestedTensorForLoopFallback() local
H A DBatchRulesReduceOps.cpp109 const auto num_arguments = schema.arguments().size(); in boxed_reduction_batch_rule() local
/aosp_15_r20/external/pytorch/torch/csrc/autograd/
H A Dautograd_not_implemented_fallback.cpp104 const auto num_arguments = schema.arguments().size(); in basicAutogradNotImplementedFallbackImpl() local
240 const auto num_arguments = schema.arguments().size(); in autogradNotImplementedFallbackImpl() local
504 const auto num_arguments = schema.arguments().size(); in autogradNotImplementedInplaceOrViewFallbackImpl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DLegacyBatchedFallback.cpp97 const auto num_arguments = static_cast<int64_t>(schema.arguments().size()); in batchedTensorInplaceForLoopFallback() local
269 const auto num_arguments = static_cast<int64_t>(schema.arguments().size()); in batchedTensorForLoopFallback() local
H A DZeroTensorFallback.cpp16 const auto num_arguments = arguments.size(); in zeroTensorFallback() local
H A DFunctionalizeFallbackKernel.cpp56 const auto num_arguments = schema.arguments().size(); in functionalizeFallback() local
H A DFunctionalTensorWrapper.cpp826 const auto num_arguments = schema.arguments().size(); in functionalize_op_helper() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DMathBitsFallback.h56 const auto num_arguments = arguments.size(); in fallback_impl() local
H A DCPUFallback.cpp96 const auto num_arguments = schema_args.size(); in cpu_fallback() local
/aosp_15_r20/external/pytorch/torch/csrc/
H A DPyInterpreter.cpp304 const auto num_arguments = schema.arguments().size(); in dispatch() local
385 const auto num_arguments = schema.arguments().size(); in python_dispatcher() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A Dbackend_fallback_test.cpp50 auto num_arguments = op.schema().arguments().size(); in generic_wrapper_fallback() local
/aosp_15_r20/external/bcc/src/cc/
H A Dbcc_usdt.h36 int num_arguments; member
H A Dusdt.h257 size_t num_arguments() const { return locations_.front().arguments_.size(); } in num_arguments() function
/aosp_15_r20/external/bcc/src/cc/includes/
H A Dbcc_usdt.h36 int num_arguments; member
H A Dusdt.h257 size_t num_arguments() const { return locations_.front().arguments_.size(); } in num_arguments() function
/aosp_15_r20/external/pytorch/aten/src/ATen/core/
H A DPythonFallbackKernel.cpp65 const auto num_arguments = schema.arguments().size(); in pythonFallback() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dtf_kernel_to_llvm_pass.cc113 auto num_arguments = arguments.size(); in generateParamsArray() local
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_eager_fallback.cpp212 const auto num_arguments = schema_args.size(); in ts_eager_fallback() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/runtime/
H A Dsymbolic_shape.cc121 size_t SymbolicShapesResolver::num_arguments() const { in num_arguments() function in xla::runtime::SymbolicShapesResolver
/aosp_15_r20/external/bcc/tests/cc/
H A Dtest_usdt_probes.cc210 int num_arguments; in probe_num_arguments() local
/aosp_15_r20/external/oj-libjdwp/src/share/javavm/export/
H A Djmm.h202 int num_arguments; /* Number of supported options or arguments */ member
/aosp_15_r20/external/flac/src/metaflac/
H A Doptions.h210 unsigned num_arguments; member
/aosp_15_r20/external/pytorch/torch/csrc/utils/
H A Dpython_dispatch.cpp144 const auto num_arguments = schema.arguments().size(); in operator ()() local
/aosp_15_r20/external/mesa3d/src/compiler/glsl/
H A Dbuiltin_functions.cpp6025 unsigned num_arguments, in add_image_function()
8806 unsigned num_arguments, in _image_prototype()
8936 unsigned num_arguments, in _image()

12