| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | triton_utils.py | 21 out_ptr, argument
38 out_ptr, argument
68 out_ptr, argument
93 out_ptr, argument
127 out_ptr, argument
163 out_ptr, argument
180 out_ptr, argument
197 out_ptr, argument
230 out_ptr, argument
249 out_ptr, argument
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_kernels.py | 275 out_ptr, argument
344 out_ptr, argument
447 out_ptr, argument
462 out_ptr, argument
496 out_ptr, argument
572 out_ptr, argument
649 out_ptr, argument
1089 out_ptr, argument
1121 out_ptr, argument
1153 out_ptr, argument
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuDirectConv2dOutputStageKernel.cpp | 132 const auto out_ptr = reinterpret_cast<T *>(out.ptr()) + x; in output_stage_nchw() local
198 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in output_stage_nhwc() local
215 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in output_stage_nhwc() local
278 const auto out_ptr = reinterpret_cast<TOut *>(out.ptr()) + x; in output_stage_nchw() local
296 const auto out_ptr = reinterpret_cast<TOut *>(out.ptr()) + x; in output_stage_nchw() local
360 const auto out_ptr = reinterpret_cast<TOut *>(out.ptr()) + x; in output_stage_nhwc() local
378 const auto out_ptr = reinterpret_cast<TOut *>(out.ptr()) + x; in output_stage_nhwc() local
|
| H A D | CpuDequantizeKernel.cpp | 130 const auto out_ptr = reinterpret_cast<TOut *>(out.ptr()); in run_dequantization_qasymm8() local
171 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in run_dequantization_qsymm8_per_channel_nchw() local
212 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in run_dequantization_qsymm8_per_channel_nhwc() local
263 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in run_dequantization_qsymm8() local
305 const auto out_ptr = reinterpret_cast<T *>(out.ptr()); in run_dequantization_qsymm16() local
|
| H A D | CpuConcatenateBatchKernel.cpp | 74 const auto out_ptr = reinterpret_cast<uint8_t *>(dst_ptr + dst_it.offset()); in batch_concat() local
95 const auto out_ptr = reinterpret_cast<int8_t *>(dst_ptr + dst_it.offset()); in batch_concat() local
114 const auto out_ptr = reinterpret_cast<T *>(dst_ptr + dst_it.offset()); in batch_concat() local
|
| H A D | CpuConcatenateDepthKernel.cpp | 77 const auto out_ptr = reinterpret_cast<uint8_t *>(dst_ptr + dst_it.offset()); in depth_concat() local
97 const auto out_ptr = reinterpret_cast<int8_t *>(dst_ptr + dst_it.offset()); in depth_concat() local
117 const auto out_ptr = reinterpret_cast<T *>(dst_ptr + dst_it.offset()); in depth_concat() local
|
| /aosp_15_r20/external/rust/android-crates-io/crates/uefi/src/proto/device_path/ |
| D | device_path_gen.rs | 3349 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3376 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3411 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3448 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3485 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3528 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3568 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3606 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3651 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3707 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
[all …]
|
| /aosp_15_r20/external/rust/crates/uefi/src/proto/device_path/ |
| D | device_path_gen.rs | 3349 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3376 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3411 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3448 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3485 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3528 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3568 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3606 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3651 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
3707 let out_ptr: *mut u8 = maybe_uninit_slice_as_mut_ptr(out); in write_data() localVariable
[all …]
|
| /aosp_15_r20/external/executorch/extension/training/optimizer/ |
| H A D | sgd.cpp | 30 auto out_ptr = out.mutable_data_ptr<float>(); in add_out_hack() local
38 auto out_ptr = out.mutable_data_ptr<float>(); in mul_out_hack() local
46 auto out_ptr = out.mutable_data_ptr<float>(); in clone_out_hack() local
|
| /aosp_15_r20/external/libtextclassifier/native/tensorflow_models/seq_flow_lite/tflite_ops/ |
| H A D | layer_norm.cc | 208 uint8_t* out_ptr = output->data.uint8; in FlexibleLayerNorm() local
298 uint8_t* out_ptr = output->data.uint8; in IntegerLayerNorm() local
334 float* out_ptr = output->data.f; in DefaultLayerNormFloat() local
366 uint8_t* out_ptr = output->data.uint8; in DefaultLayerNorm() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/ |
| H A D | impl.cpp | 53 const auto out_ptr = reinterpret_cast<T *>(output.ptr()); in neon_logits_1d_max() local
113 const auto out_ptr = reinterpret_cast<T *>(out_it.ptr()) + start_x; in neon_softmax_logits_1d_quantized() local
288 const auto out_ptr = reinterpret_cast<T *>(out_it.ptr()) + start_x; in neon_softmax_logits_1d_float() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| H A D | batch_matmul.h | 104 float* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
223 float* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
361 int8_t* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
467 int32_t* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/indirect-interleaves/ |
| H A D | a64_interleave8_block1_s16_s16.hpp | 29 int16_t * &out_ptr, const int16_t * const * in, size_t width, size_t height, in interleave_block()
271 uint16_t * &out_ptr, const uint16_t * const * in, size_t width, size_t height, in interleave_block()
|
| H A D | a64_interleave8_block4_s8_s8.hpp | 29 int8_t * &out_ptr, const int8_t * const * in, size_t width, size_t height, in interleave_block()
332 uint8_t * &out_ptr, const uint8_t * const * in, size_t width, size_t height, in interleave_block()
|
| H A D | a64_interleave4_block16_s8_s8.hpp | 29 int8_t * &out_ptr, const int8_t * const * in, size_t width, size_t height, in interleave_block()
182 uint8_t * &out_ptr, const uint8_t * const * in, size_t width, size_t height, in interleave_block()
|
| H A D | a64_interleave8_block8_s8_s8.hpp | 29 int8_t * &out_ptr, const int8_t * const * in, size_t width, size_t height, in interleave_block()
308 uint8_t * &out_ptr, const uint8_t * const * in, size_t width, size_t height, in interleave_block()
|
| /aosp_15_r20/external/ComputeLibrary/tests/ |
| H A D | AssetsLibrary.h | 681 … const auto out_ptr = static_cast<RawTensor::value_type *>(tensor(id)); in fill() local
702 … const auto out_ptr = static_cast<RawTensor::value_type *>(tensor(id)); in fill() local
715 … const auto out_ptr = static_cast<RawTensor::value_type *>(tensor(id)); in fill() local
|
| /aosp_15_r20/external/libgav1/src/ |
| H A D | decoder.cc | 66 Libgav1Decoder* decoder, const Libgav1DecoderBuffer** out_ptr) { in Libgav1DecoderDequeueFrame()
102 StatusCode Decoder::DequeueFrame(const DecoderBuffer** out_ptr) { in DequeueFrame()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | WeightNormKernel.cpp | 52 scalar_t* out_ptr, in sum_norm_per_row()
65 float* out_ptr, in sum_norm_per_row()
226 scalar_t* out_ptr, in sum_product_per_row()
241 float* out_ptr, in sum_product_per_row()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/gemm_matrix_add/generic/neon/ |
| H A D | impl.cpp | 50 const auto out_ptr = reinterpret_cast<float *>(out.ptr()); in matrix_addition_f32() local
95 const auto out_ptr = reinterpret_cast<float16_t *>(out.ptr()); in matrix_addition_f16() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | batch_matmul.h | 90 Tout* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
176 float* out_ptr = output_data + ((b0 * batch_dim1 * batch_dim2) + in BatchMatMul() local
246 T* out_ptr = output_data + in BatchMatMul() local
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/reference/ |
| H A D | Convolution3d.h | 57 T *out_ptr = out.data() + o_offset; variable
107 T *out_ptr = out.data() + o_offset; variable
|
| H A D | ROIPoolingLayer.cpp | 104 …auto out_ptr = output.data() + px + py * output_shape[0] + fm * output_shape[0] * output_shape[1] … in roi_pool_layer() local
121 …auto out_ptr = output.data() + px + py * output_shape[0] + fm * output_shape[0] * output_shape[1] … in roi_pool_layer() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/l2normlayer/generic/neon/ |
| H A D | impl.cpp | 56 const auto out_ptr = reinterpret_cast<T *>(output_it.ptr()); in l2_normalize_x() local
103 const auto out_ptr = reinterpret_cast<T *>(output_it.ptr()); in l2_normalize_yz() local
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/softmax/generic/sve/ |
| H A D | impl.cpp | 48 const auto out_ptr = reinterpret_cast<ScalarType *>(output.ptr()); in sve_logits_1d_max() local
89 const auto out_ptr = reinterpret_cast<ScalarType *>(out_it.ptr()) + start_x; in sve_softmax_logits_1d_float() local
|