| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | group_norm_kernel.cpp | 103 Vec x_vec = Vec::loadu(X_ptr + d); in ColumnwiseMoments() local
108 Vec x_vec = Vec::loadu(X_ptr + d, D - d); in ColumnwiseMoments() local
954 Vec x_vec = Vec::loadu(X_ptr + d); in DsDbRowwiseMomentsChannelsLast() local
965 Vec x_vec = Vec::loadu(X_ptr + d, C - d); in DsDbRowwiseMomentsChannelsLast() local
990 Vec x_vec = Vec::loadu(X_ptr + d); in DsDbRowwiseMomentsChannelsLast() local
1010 Vec x_vec = Vec::loadu(X_ptr + d, C - d); in DsDbRowwiseMomentsChannelsLast() local
1076 auto x_vec = vec::Vectorized<T>::loadu(X_ptr + d); in ApplyInputGradientsChannelsLastColMov() local
1091 auto x_vec = vec::Vectorized<T>::loadu(X_ptr + d, D - d); in ApplyInputGradientsChannelsLastColMov() local
1128 Vec x_vec = Vec::loadu(X_ptr + d); in ApplyInputGradientsChannelsLastColMov() local
1146 Vec x_vec = Vec::loadu(X_ptr + d, D - d); in ApplyInputGradientsChannelsLastColMov() local
[all …]
|
| H A D | Activation.cpp | 388 [&](Vec x_vec) { in GeluKernelImpl()
429 [&](Vec x_vec) { in GeluKernelImpl()
470 [&](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in GeluBackwardKernelImpl()
533 [&](Vec dy_vec, Vec x_vec) { in GeluBackwardKernelImpl()
605 [&](Vec dy_vec, Vec x_vec) { in GeluBackwardKernelImpl()
837 [&](vec::Vectorized<scalar_t> x_vec) { in hardswish_kernel()
864 [&](Vec x_vec) { in hardswish_kernel()
1230 [kOneVec](Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in silu_kernel()
1246 [kOneVec](Vectorized<scalar_t> x_vec) { in silu_kernel()
1264 [kOneVec](Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) -> Vectorized<scalar_t> { in silu_backward_kernel()
[all …]
|
| H A D | UnaryOpsKernel.cpp | 79 [](Vectorized<VT> x_vec) { return x_vec.log(); }, in VmlLog()
156 [kOneVec](Vectorized<scalar_t> x_vec) { in logit_kernel()
173 [kOneVec, lo_vec, hi_vec](Vectorized<scalar_t> x_vec) { in logit_kernel()
|
| H A D | moments_utils.h | 68 const Vec x_vec = Vec::loadu(X_ptr + j * Vec::size()); in UpdateMomentsVec() local
|
| H A D | BinaryOpsKernel.cpp | 875 Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) { in logit_backward_kernel()
896 Vectorized<scalar_t> dy_vec, Vectorized<scalar_t> x_vec) { in logit_backward_kernel()
|
| /aosp_15_r20/external/piex/src/binary_parse/ |
| H A D | range_checked_byte_ptr.cc | 265 std::vector<unsigned char> x_vec = x.extractBytes(0, num); in memcmp() local
277 std::vector<unsigned char> x_vec = x.extractBytes(0, y.length()); in strcmp() local
|
| /aosp_15_r20/external/sdv/vsomeip/third_party/boost/numeric/odeint/examples/ |
| D | harmonic_oscillator.cpp | 116 vector<state_type> x_vec; in main() local
|
| /aosp_15_r20/external/executorch/kernels/optimized/cpu/ |
| H A D | moments_utils.h | 78 const Vec x_vec = Vec::loadu(X_ptr + j * Vec::size()); in UpdateMomentsVec() local
|
| /aosp_15_r20/external/webrtc/modules/audio_coding/neteq/tools/ |
| H A D | neteq_delay_analyzer.cc | 39 const std::vector<int64_t>& x_vec, in LinearInterpolate()
|
| /aosp_15_r20/external/pytorch/test/cpp/api/ |
| H A D | serialize.cpp | 981 std::vector<torch::Tensor> x_vec = { in TEST() local
|
| /aosp_15_r20/external/libvpx/third_party/libyuv/source/ |
| H A D | scale_msa.cc | 641 v4i32 x_vec = __msa_fill_w(x); in ScaleARGBCols_MSA() local
|
| /aosp_15_r20/external/libyuv/source/ |
| H A D | scale_msa.cc | 641 v4i32 x_vec = __msa_fill_w(x); in ScaleARGBCols_MSA() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vec_test_all_types.cpp | 1758 auto x_vec = vec::loadu(x); in TYPED_TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | constant_folding_test.cc | 540 Output x_vec = ops::Placeholder(s.WithOpName("x_vec"), DT_FLOAT, in TEST_F() local
|