Home
last modified time | relevance | path

Searched defs:v_output (Results 1 – 25 of 30) sorted by relevance

12

/external/pytorch/aten/src/ATen/native/vulkan/ops/
DSelect.cpp18 vTensor v_output{ in select_batch_4d() local
70 vTensor v_output{ in select_depth_3d() local
117 vTensor v_output{ in select_depth_4d() local
170 vTensor v_output{ in select_height_3d() local
229 vTensor v_output{ in select_height_4d() local
283 vTensor v_output{ in select_width_3d() local
343 vTensor v_output{ in select_width_4d() local
DConcat.cpp19 Tensor cat_batch(const MaterializedITensorListRef& tensors, vTensor& v_output) { in cat_batch()
56 vTensor& v_output) { in cat_feature()
145 vTensor& v_output) { in cat_feature_mult4ch()
199 Tensor cat_width(const MaterializedITensorListRef& tensors, vTensor& v_output) { in cat_width()
237 vTensor& v_output) { in cat_height()
308 vTensor v_output{ in cat() local
DSlice.cpp21 vTensor& v_output) { in slice_4d()
87 vTensor& v_output) { in slice_width()
163 vTensor& v_output) { in slice_height()
280 vTensor v_output{ in slice() local
DUpsample.cpp27 vTensor v_output{ in upsample_nearest2d() local
112 vTensor v_output{ in upsample_bilinear2d() local
DPermute.cpp17 vTensor& v_output) { in permute_4d()
106 vTensor v_output{ in permute() local
DTranspose.cpp18 vTensor& v_output) { in transpose_4d()
125 vTensor v_output{ in transpose() local
DQuantizedTensor.cpp43 vTensor v_output{ in quantize_per_tensor() local
118 vTensor v_output{ in dequantize_helper() local
DBatchnorm.cpp20 vTensor& v_output, in record_op()
223 vTensor v_output{ in run_batchnorm_context() local
DZero.cpp56 vTensor v_output{ in zeros() local
DPool.cpp25 vTensor v_output{ in adaptive_avg_pool2d() local
154 vTensor v_output{ in pool2d() local
DGlu.cpp31 vTensor v_output{ in glu() local
DShape.cpp20 vTensor v_output{ in view_internal() local
DFlip.cpp26 vTensor v_output{ in flip() local
DUnsqueeze.cpp44 vTensor v_output{ in unsqueeze() local
DMean.cpp54 vTensor v_output{ in mean_dim() local
DPadding.cpp52 vTensor v_output{ in pad2d() local
DSum.cpp43 vTensor v_output{ in sum_dim() local
DUnaryOp.cpp22 vTensor v_output{ in unary_op() local
Dcumsum.cpp111 vTensor v_output{ in cumsum() local
DLerp.cpp59 vTensor v_output{ in _lerp_scalar() local
184 vTensor v_output{ in _lerp_tensor() local
DBinaryOp.cpp25 vTensor v_output{ in binary_op_scalar() local
167 vTensor v_output{ in binary_op_tensor() local
242 vTensor v_output{ in quantized_binary_op_tensor() local
DClamp.cpp24 vTensor v_output{ in _clamp() local
202 vTensor v_output{ in activation() local
341 vTensor v_output{ in activation_scalar() local
DSoftmax.cpp107 vTensor v_output{ in softmax_internal() local
DMm.cpp465 vTensor v_output{ in run_quantized_addmm_context() local
692 vTensor v_output{ in run_addmm_context() local
817 vTensor v_output{ in run_baddbmm_context() local
/external/libultrahdr/lib/src/dsp/arm/
Dgainmapmath_neon.cpp109 const int16x8_t v_output = vcombine_s16(vqrshrn_n_s32(v_lo, 14), vqrshrn_n_s32(v_hi, 14)); in vConversion_neon() local
116 const int16x8_t v_output = vConversion_neon(u, v, coeffs); in yuvConversion_neon() local
161 const uint8x8_t v_output = vqmovun_s16(vaddq_s16(new_v, vdupq_n_s16(128))); in transformYuv420_neon() local
222 const uint8x16_t v_output = vcombine_u8(v_output_lo, v_output_hi); in transformYuv444_neon() local

12