Home
last modified time | relevance | path

Searched defs:nInputs (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.h27 int nInputs, nOutputs; member
44 int nInputs, in make_kernel_descriptor()
H A DCUDAJitLoops.cuh275 constexpr int nInputs = arity; in jitted_gpu_kernel_impl() local
H A Djit_utils.cpp971 int nInputs, in generate_code()
H A DReduce.cuh1342 constexpr int nInputs = 1; in jitted_gpu_reduce_kernel() local
/aosp_15_r20/external/pdfium/third_party/lcms/src/
H A Dcmslut.c748 cmsUInt32Number nInputs, nOutputs; in cmsStageSampleCLut16bit() local
814 cmsUInt32Number nInputs, nOutputs; in cmsStageSampleCLutFloat() local
870 cmsBool CMSEXPORT cmsSliceSpace16(cmsUInt32Number nInputs, const cmsUInt32Number clutPoints[], in cmsSliceSpace16()
901 cmsInt32Number CMSEXPORT cmsSliceSpaceFloat(cmsUInt32Number nInputs, const cmsUInt32Number clutPoin… in cmsSliceSpaceFloat()
H A Dcmsopt.c52 cmsUInt32Number nInputs; member
395 cmsUInt32Number nInputs, cmsToneCurve** In, in PrelinOpt16alloc()
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A Djiterator.cu31 int nInputs = iter.ninputs(); in launch_jitted_vectorized_kernel_dynamic() local
128 int nInputs = iter.ninputs(); in launch_jitted_unrolled_kernel_dynamic() local
/aosp_15_r20/external/pdfium/core/fpdfapi/page/
H A Dcpdf_function.cpp101 size_t nInputs = m_nInputs * 2; in Init() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dkernel_spec.h91 int64_t nInputs() const { in nInputs() function
/aosp_15_r20/external/armnn/shim/sl/canonical/
H A DConversionUtils.hpp539 size_t nInputs = inputs.size(); in SwizzleInputs() local
564 size_t nInputs = inputs.size(); in TransposeInputTensors() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dtensorexpr_fuser.cpp1172 auto const nInputs = consumer->inputs().size() + in canMerge() local
/aosp_15_r20/external/pdfium/third_party/lcms/include/
H A Dlcms2_plugin.h297 cmsUInt32Number nInputs; // != 1 only in 3D interpolation member
/aosp_15_r20/external/android-nn-driver/
H A DConversionUtils.hpp601 size_t nInputs = inputs.size(); in SwizzleInputs() local
626 size_t nInputs = inputs.size(); in TransposeInputTensors() local