Home
last modified time | relevance | path

Searched defs:tmp_output (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DFunctionalizeFallbackKernel.cpp145 at::Tensor tmp_output; in resize__functionalization() local
296 at::Tensor tmp_output; in _unsafe_view_functionalize() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEReduceMean.cpp119 ITensor *tmp_output = output; in configure() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLReduceMean.cpp131 ICLTensor *tmp_output = output; in configure() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DRreluWithNoise.cu80 Tensor tmp_output = output.contiguous(); in _rrelu_with_noise_cuda_train() local
H A DMultiMarginLoss.cu256 auto tmp_output = at::empty({nframe}, input.options()); in multi_margin_loss_cuda_out() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/
H A Ddraw_gs.h65 float (*tmp_output)[4]; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkldnn/
H A DPooling.cpp526 auto tmp_output = at::native::mkldnn_adaptive_avg_pool2d(input, output_size); in mkldnn_adaptive_avg_pool2d_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DReduceOps.cpp1214 auto tmp_output = at::empty(result.sizes(), result.options().dtype(kFloat)); in TORCH_IMPL_FUNC() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc2761 auto tmp_output = in HandleReshape() local