Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cuda/
Djit_utils.h27 int nInputs, nOutputs; member
44 int nInputs, in make_kernel_descriptor()
DCUDAJitLoops.cuh275 constexpr int nInputs = arity; in jitted_gpu_kernel_impl() local
Djit_utils.cpp971 int nInputs, in generate_code()
DReduce.cuh1342 constexpr int nInputs = 1; in jitted_gpu_reduce_kernel() local
/external/pdfium/core/fpdfapi/page/
Dcpdf_function.cpp101 size_t nInputs = m_nInputs * 2; in Init() local
/external/pytorch/aten/src/ATen/cuda/
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
/external/pytorch/torch/csrc/jit/codegen/fuser/
Dkernel_spec.h91 int64_t nInputs() const { in nInputs() function
/external/pdfium/third_party/lcms/src/
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()
Dcmsopt.c52 cmsUInt32Number nInputs; member
395 cmsUInt32Number nInputs, cmsToneCurve** In, in PrelinOpt16alloc()
/external/armnn/shim/sl/canonical/
DConversionUtils.hpp539 size_t nInputs = inputs.size(); in SwizzleInputs() local
564 size_t nInputs = inputs.size(); in TransposeInputTensors() local
/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp1172 auto const nInputs = consumer->inputs().size() + in canMerge() local
/external/pdfium/third_party/lcms/include/
Dlcms2_plugin.h297 cmsUInt32Number nInputs; // != 1 only in 3D interpolation member
/external/android-nn-driver/
DConversionUtils.hpp601 size_t nInputs = inputs.size(); in SwizzleInputs() local
626 size_t nInputs = inputs.size(); in TransposeInputTensors() local