Home
last modified time | relevance | path

Searched defs:vec_size (Results 1 – 25 of 47) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
H A Dutil.cc29 DataType dst_type, int vec_size) { in GetGlslConversion()
219 int vec_size) { in GetTypeDeclaration()
233 int vec_size) { in GetZeroValue()
248 int vec_size) { in GetOneValue()
263 DataType dst_type, int vec_size) { in GetTypeConversion()
H A Dtensor_desc.cc388 const int vec_size = const_expr == "scalar_type" ? 1 : 4; in PerformConstExpr() local
392 const int vec_size = const_expr == "scalar_zero_value" ? 1 : 4; in PerformConstExpr() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Ddata_type.cc28 const std::string& vec_type, int vec_size) { in ToGlslType()
109 std::string ToCLDataType(DataType data_type, int vec_size) { in ToCLDataType()
142 std::string ToMetalDataType(DataType data_type, int vec_size) { in ToMetalDataType()
194 std::string ToGlslShaderDataType(DataType data_type, int vec_size, in ToGlslShaderDataType()
/aosp_15_r20/external/executorch/extension/llm/custom_ops/spinquant/
H A Dfast_hadamard_transform.cpp29 int vec_size) { in quantized_normalize_after_fht()
58 const int vec_size = 1 << log2_vec_size; in fast_hadamard_transform_symmetric_quantized_s16() local
84 const int vec_size = (1 << log2_vec_size); in fast_hadamard_transform_symmetric_quantized_s16_28N() local
H A Dfast_hadamard_transform.h40 const int vec_size = 1 << log2_vec_size; in normalize_after_fht() local
58 const auto vec_size = 1 << log2_vec_size; in fast_hadamard_transform_unnormalized_simple_impl() local
128 const int vec_size = (1 << log2_vec_size); in fast_hadamard_transform_28N() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Djit_utils.cpp942 int vec_size, in generate_code()
983 int vec_size, in generate_code()
1309 int vec_size, in generate_reduction_code()
1339 int vec_size, in generate_reduction_code()
H A DDropout.cu203 int vec_size = 4; in get_vector_size() local
249 int vec_size = get_vector_size<scalar_t>(self, ret, mask); in launcher() local
/aosp_15_r20/external/linux-kselftest/tools/testing/selftests/mincore/
H A Dmincore_selftest.c190 int vec_size; in TEST() local
295 int vec_size; in TEST() local
/aosp_15_r20/external/libbrillo/brillo/dbus/
H A Ddata_serialization_fuzzer.cc159 int vec_size = data_provider.ConsumeIntegralInRange<int>( in LLVMFuzzerTestOneInput() local
168 int vec_size = data_provider.ConsumeIntegralInRange<int>( in LLVMFuzzerTestOneInput() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/softmax/generic/neon/
H A Dimpl.cpp107 constexpr int vec_size = 16; in neon_softmax_logits_1d_quantized() local
281 constexpr int vec_size = 16 / sizeof(T); in neon_softmax_logits_1d_float() local
/aosp_15_r20/external/libaom/aom_dsp/
H A Dfft.c61 aom_fft_unpack_func_t unpack, int vec_size) { in aom_fft_2d_gen()
118 aom_fft_transpose_func_t transpose, int vec_size) { in aom_ifft_2d_gen()
/aosp_15_r20/external/libaom/av1/common/x86/
H A Dav1_txfm_sse4.c20 const int vec_size = size >> 2; in av1_round_shift_array_sse4_1() local
/aosp_15_r20/external/executorch/extension/llm/custom_ops/spinquant/test/
H A Dfast_hadamard_transform_test_impl.h25 const int vec_size = (1 << log2_vec_size); in fast_hadamard_transform_28N_with_transpose() local
/aosp_15_r20/external/skia/src/utils/
H A DSkJSON.cpp92 static void* MakeVector(size_t vec_size, const void* src, size_t src_size, SkArenaAlloc& alloc) { in MakeVector()
104 static void* MakeVector(size_t vec_size, const void* src, SkArenaAlloc& alloc) { in MakeVector()
/aosp_15_r20/external/OpenCL-CTS/test_conformance/spirv_new/
H A Dtest_op_vector_times_scalar.cpp52 int vec_size = sizeof(Tv) / sizeof(Ts); in test_vector_times_scalar() local
/aosp_15_r20/external/executorch/extension/llm/custom_ops/
H A Dop_sdpa.cpp91 auto vec_size = vec::Vectorized<T1>::size(); in _exp_reduce_sum_fusion_kernel() local
126 auto vec_size = vec::Vectorized<scalar_t>::size(); in _mul_reduce_max_fusion_kernel() local
/aosp_15_r20/external/mesa3d/src/compiler/glsl/
H A Dserialize.cpp482 unsigned vec_size = in write_uniforms() local
549 unsigned vec_size = in read_uniforms() local
/aosp_15_r20/device/google/cuttlefish/guest/hals/light/lights/
Dlights_vsock_server.rs105 let vec_size = buffer.len() as u32; in send_buffer_with_length() localVariable
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/template_writer/cl/
H A DClTemplatePool2d.cpp462 …const unsigned int vec_size = adjust_vec_size(((_dst->data_type() == DataType::F32) ? 2 : 4), … in get_window() local
/aosp_15_r20/packages/modules/Bluetooth/system/gd/packet/parser/fields/
Dvector_field.cc249 std::string vec_size = accessor + ".size()"; in GenStringRepresentation() local
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLPadLayerKernel.cpp98 …const unsigned int vec_size = adjust_vec_size(std::min(16U, 32U / static_cast<unsig… in configure() local
H A DCLReductionOperationKernel.cpp117 …unsigned int vec_size = (is_data_type_quantized(input->info()->data_type()) && (axi… in configure() local
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClScaleKernel.cpp121 unsigned int vec_size = 0; in configure() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DFlashAttentionKernel.cpp60 auto vec_size = vec::Vectorized<T1>::size(); in _exp_reduce_sum_fusion_kernel() local
95 auto vec_size = vec::Vectorized<scalar_t>::size(); in _mul_reduce_max_fusion_kernel() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dconv_serialization.h232 int64_t vec_size = params_vec.size(); in serialize_conv() local

12