| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/ |
| H A D | kernel_backward.h | 629 struct Params { struct
631 const scalar_t* query_ptr = nullptr; // [Mq, nH, K]
632 const scalar_t* key_ptr = nullptr; // [Mk, nH, K]
633 const scalar_t* value_ptr = nullptr; // [Mk, nH, Kv]
634 const scalar_t* bias_ptr = nullptr;
635 const lse_scalar_t* logsumexp_ptr = nullptr; // [nH, Mq]
636 const scalar_t* output_ptr = nullptr; // [Mq, nH, Kv]
637 const scalar_t* grad_output_ptr = nullptr; // [Mq, nH, Kv]
638 accum_t* delta_ptr = nullptr; // [nH, Mq]
639 const int32_t* cu_seqlens_q_ptr = nullptr;
[all …]
|
| H A D | kernel_forward.h | 132 struct Params { struct
134 const scalar_t* query_ptr = nullptr; // [num_queries, num_heads, head_dim]
135 const scalar_t* key_ptr = nullptr; // [num_keys, num_heads, head_dim]
136 const scalar_t* value_ptr = nullptr; // [num_keys, num_heads, head_dim_value]
137 const scalar_t* attn_bias_ptr = nullptr; // [num_heads, num_queries, num_keys]
138 const int32_t* seqstart_q_ptr = nullptr;
139 const int32_t* seqstart_k_ptr = nullptr;
141 const int32_t* seqlen_k_ptr = nullptr;
142 uint32_t causal_diagonal_offset = 0;
145 output_t* output_ptr = nullptr; // [num_queries, num_heads, head_dim_value]
[all …]
|
| /aosp_15_r20/external/armnn/tests/ |
| H A D | InferenceModel.hpp | 61 struct Params struct
63 std::string m_ModelPath;
64 std::vector<std::string> m_InputBindings;
65 std::vector<armnn::TensorShape> m_InputShapes;
66 std::vector<std::string> m_OutputBindings;
67 std::vector<armnn::BackendId> m_ComputeDevices;
68 std::string m_DynamicBackendsPath;
69 size_t m_SubgraphId;
70 bool m_AllowExpandedDims;
71 bool m_IsModelBinary;
[all …]
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/ |
| H A D | fpA_intB_gemm.h | 157 struct Params struct
159 cutlass::gemm::GemmCoord problem_size;
160 cutlass::gemm::GemmCoord grid_tiled_shape;
161 int swizzle_log_tile;
162 typename Mma::IteratorA::Params params_A;
163 typename Mma::IteratorA::TensorRef ref_A;
164 typename Mma::IteratorB::Params params_B;
165 typename Mma::IteratorB::TensorRef ref_B;
166 typename Mma::IteratorScale::Params params_scale;
167 typename Mma::IteratorScale::TensorRef ref_scale;
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | dataset.h | 394 struct Params { struct
395 explicit Params(IteratorContext* ctx) in Params() function
414 explicit Params(OpKernelContext* ctx) in Params() function
444 std::function<Allocator*(AllocatorAttributes)> allocator_getter = nullptr;
447 CancellationManager* cancellation_manager;
450 CollectiveExecutor* collective_executor = nullptr;
453 Env* env = nullptr;
456 FunctionLibraryRuntime* flr = nullptr;
459 FunctionHandleCache* function_handle_cache = nullptr;
464 int64 interleave_depth = 0;
[all …]
|
| H A D | op_kernel.h | 562 struct Params { struct
563 ~Params() { delete eigen_gpu_device; } in ~Params() argument
586 // ensure_eigen_gpu_device, and then if the Params structure is argument
590 PerOpGpuDevice* eigen_gpu_device = nullptr;
592 inline void ensure_eigen_gpu_device() { in ensure_eigen_gpu_device()
606 bool track_allocations = false;
607 bool log_memory = false;
610 const AllocatorAttributes* output_attr_array = nullptr;
613 ResourceMgr* resource_manager = nullptr;
617 ScopedStepContainer* step_container = nullptr;
[all …]
|
| /aosp_15_r20/external/crosvm/devices/src/virtio/video/ |
| H A D | params.rs | 24 pub struct Params { struct
38 impl TryFrom<virtio_video_params> for Params { argument
83 impl TryFrom<virtio_video_params_ext> for Params { implementation
104 impl Params { impl
106 let Params { in to_virtio_video_params() localVariable
141 let Params { resource_type, .. } = self; in to_virtio_video_params_ext() localVariable
|
| /aosp_15_r20/external/ruy/example/ |
| H A D | parametrized_example.cc | 44 struct Params { struct
64 void run(const Params& params) { in run() argument
|
| /aosp_15_r20/device/google/zumapro/ |
| D | task_profiles.json | 8 "Params": object
23 "Params": object
38 "Params": object
53 "Params": object
68 "Params": object
83 "Params": object
96 "Params": object
109 "Params": object
122 "Params": object
137 "Params": object
[all …]
|
| /aosp_15_r20/device/google/gs101/ |
| D | task_profiles.json | 8 "Params": object
23 "Params": object
38 "Params": object
53 "Params": object
68 "Params": object
83 "Params": object
96 "Params": object
109 "Params": object
122 "Params": object
137 "Params": object
[all …]
|
| /aosp_15_r20/device/google/gs201/ |
| D | task_profiles.json | 8 "Params": object
23 "Params": object
38 "Params": object
53 "Params": object
68 "Params": object
83 "Params": object
96 "Params": object
109 "Params": object
122 "Params": object
137 "Params": object
[all …]
|
| /aosp_15_r20/device/google/zuma/ |
| D | task_profiles.json | 8 "Params": object
23 "Params": object
38 "Params": object
53 "Params": object
68 "Params": object
83 "Params": object
96 "Params": object
109 "Params": object
122 "Params": object
137 "Params": object
[all …]
|
| /aosp_15_r20/out/target/product/shiba/obj/ETC/task_profiles.json_intermediates/ |
| D | task_profiles.json | 93 "Params": object
106 "Params": object
119 "Params": object
132 "Params": object
145 "Params": object
158 "Params": object
171 "Params": object
184 "Params": object
197 "Params": object
210 "Params": object
[all …]
|
| /aosp_15_r20/system/core/libprocessgroup/profiles/ |
| H A D | task_profiles.json | 93 "Params": object
106 "Params": object
119 "Params": object
132 "Params": object
145 "Params": object
158 "Params": object
171 "Params": object
184 "Params": object
197 "Params": object
210 "Params": object
[all …]
|
| /aosp_15_r20/out/soong/.intermediates/packages/modules/Virtualization/build/microdroid/microdroid/android_common/root/system/etc/ |
| D | task_profiles.json | 93 "Params": object
106 "Params": object
119 "Params": object
132 "Params": object
145 "Params": object
158 "Params": object
171 "Params": object
184 "Params": object
197 "Params": object
210 "Params": object
[all …]
|
| /aosp_15_r20/out/soong/.intermediates/system/core/libprocessgroup/profiles/task_profiles.json/android_arm64_armv8-2a_cortex-a55/ |
| D | task_profiles.json | 93 "Params": object
106 "Params": object
119 "Params": object
132 "Params": object
145 "Params": object
158 "Params": object
171 "Params": object
184 "Params": object
197 "Params": object
210 "Params": object
[all …]
|
| /aosp_15_r20/out/target/product/shiba/system/etc/ |
| D | task_profiles.json | 93 "Params": object
106 "Params": object
119 "Params": object
132 "Params": object
145 "Params": object
158 "Params": object
171 "Params": object
184 "Params": object
197 "Params": object
210 "Params": object
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
49 const void *weights; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
54 int8_t *const *const outptrs; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
55 const int8_t *inptrs[25]; in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
57 Params( in a64_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
49 const int8_t *weights; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
54 int8_t *const *const outptrs; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
55 const int8_t *inptrs[36]; in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
57 Params( in sve_s8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
49 const void *weights; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
54 uint8_t *const *const outptrs; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
55 const uint8_t *inptrs[25]; in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
57 Params( in a64_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
49 const int8_t *weights; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
54 uint8_t *const *const outptrs; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
55 const uint8_t *inptrs[25]; in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
57 Params( in sve_u8s8u8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
49 const int8_t *weights; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
54 int8_t *const *const outptrs; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
55 const int8_t *inptrs[16]; in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
57 Params( in sve_s8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
49 const int8_t *weights; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
54 int8_t *const *const outptrs; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
55 const int8_t *inptrs[25]; in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl()
57 Params( in sve_s8q_nhwc_3x3_s2_output2x2_mla_depthfirst_impl() function
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
49 const int8_t *weights; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
54 uint8_t *const *const outptrs; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
55 const uint8_t *inptrs[16]; in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl()
57 Params( in sve_u8s8u8q_nhwc_3x3_s1_output2x2_mla_depthfirst_impl() argument
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst/ |
| H A D | generic.cpp | 46 struct Params in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl() struct
48 long unsigned int n_channels; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
49 const uint8_t *weights; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
50 const int32_t *bias; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
51 const arm_gemm::Requantize32 *requant; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
52 const int32_t *const requant_muls; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
53 const int32_t *const requant_shifts; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
54 uint8_t *const *const outptrs; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
55 const uint8_t *inptrs[36]; in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl()
57 Params( in sve_u8q_nhwc_5x5_s1_output2x2_mla_depthfirst_impl() function
|