Home
last modified time | relevance | path

Searched defs:output_permuted (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuSoftmax.cpp147 TensorInfo output_permuted(dst->clone()->set_tensor_shape(permuted_shape)); in validate() local
169 …CpuAuxTensorHandler output_permuted(offset_int_vec(InternalTensorIdx::PERMUTED_DST), _output_permu… in run() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClSoftmax.cpp114 TensorInfo output_permuted(dst.clone()->set_tensor_shape(permuted_shape)); in validate() local