Home
last modified time | relevance | path

Searched defs:b_sizes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/test/cpp/jit/
H A Dtest_irparser.cpp290 auto b_sizes = *b_type->sizes().concrete_sizes(); local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DConvolutionMM2d.cu101 const auto b_sizes = bias.sizes(); in slow_conv2d_shape_check() local
/aosp_15_r20/external/executorch/kernels/test/
H A Dop_mul_test.cpp142 for (const auto& b_sizes : b_sizeses) { in test_broadcast_a2b() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DMm.cpp183 const IntArrayRef b_sizes = bias.sizes(); in pack_biases_quantized_weights() local