Home
last modified time | relevance | path

Searched defs:bVec (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/deqp/external/openglcts/modules/common/
H A DglcLimitTest.inl266 tcu::BVec3 bVec = tcu::equal(a, b); local
279 tcu::BVec3 bVec = tcu::greaterThan(a, b); local
292 tcu::BVec3 bVec = tcu::lessThan(a, b); local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DWeightNormKernel.cpp68 using bVec = vec::Vectorized<BFloat16>; in sum_norm_per_row() typedef
106 using bVec = vec::Vectorized<BFloat16>; in apply_norm_per_row() typedef
245 using bVec = vec::Vectorized<BFloat16>; in sum_product_per_row() typedef
292 using bVec = vec::Vectorized<BFloat16>; in apply_per_row_backward() typedef
H A DLerpKernel.cpp65 using bVec = Vectorized<BFloat16>; in lerp_scalar_kernel() typedef
116 using bVec = Vectorized<BFloat16>; in lerp_tensor_kernel() typedef
H A Dbatch_norm_kernel.cpp714 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_contiguous_impl() typedef
778 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_channels_last_impl() typedef
833 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_collect_stats_contiguous_internal() typedef
906 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_collect_stats_channels_last_internal() typedef
1002 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_backward_contiguous_internal() typedef
1126 using bVec = Vectorized<scalar_t>; in batch_norm_cpu_backward_channels_last_internal() typedef
H A Dlayer_norm_kernel.cpp97 using bVec = Vectorized<T>; in layer_norm_kernel_mixed_type() typedef
316 using bVec = Vectorized<T>; in layer_norm_backward_frame() typedef
H A DUpSampleMoreKernel.cpp38 using bVec = Vectorized<scalar_nonconst_t>; in nearest_channels_last_acc() typedef
75 using bVec = Vectorized<scalar_nonconst_t>; in linear_channels_last_acc() typedef
H A DAdaptiveAvgPoolKernel.cpp177 using bVec = vec::Vectorized<scalar_t>; in cpu_adaptive_avg_pool2d_channels_last() typedef
594 using bVec = vec::Vectorized<scalar_t>; in cpu_adaptive_avg_pool3d_channels_last() typedef
H A DAdaptiveMaxPoolKernel.cpp226 using bVec = vec::Vectorized<scalar_t>; in cpu_adaptive_max_pool2d_channels_last() typedef
710 using bVec = vec::Vectorized<BFloat16>; in cpu_adaptive_max_pool3d_channels_last() typedef
H A DAvgPoolKernel.cpp242 using bVec = vec::Vectorized<scalar_t>; in cpu_avg_pool2d_channels_last() typedef
795 using bVec = vec::Vectorized<BFloat16>; in cpu_avg_pool3d_channels_last() typedef
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
H A Dbroadcast_propagation.cc119 SmallVector<int64_t> bVec = in composeBroadcastDimensionsAttr() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DUpSample.h487 using bVec = Vectorized<scalar_out>; in apply_grad_input() local
H A DEmbeddingBag.cpp265 using bVec = vec::Vectorized<BFloat16>; in index_select_add() typedef
648 using bVec = vec::Vectorized<BFloat16>; in index_select_scale_add() typedef
/aosp_15_r20/external/angle/src/common/
H A Dmatrix_utils_unittest.cpp241 Mat4 bVec(elementsVector); in TEST() local