Searched defs:bias_sizes (Results 1 – 4 of 4) sorted by relevance
185 at::OptionalIntArrayRef bias_sizes = std::nullopt; in unpack_quantized_prepacked_sizes_conv2d() local
165 int64_t bias_sizes[1] = {4}; in TEST_F() local
264 at::OptionalIntArrayRef bias_sizes, in compute_shape_convolution_backward()
1347 …auto bias_sizes = bias.defined() ? std::optional<SymIntArrayRef>(bias.sym_sizes()) : bias_sizes_op… in select_conv_backend() local