Searched refs:empty_with_tail_padding (Results 1 – 9 of 9) sorted by relevance
11 Tensor empty_with_tail_padding( in empty_with_tail_padding() function55 Tensor padded_input = empty_with_tail_padding( in allocate_padded_contiguous_if_needed()
16 at::Tensor empty_with_tail_padding(
70 Tensor output = mobile::empty_with_tail_padding( in hardswish()89 Tensor output = mobile::empty_with_tail_padding( in hardswish_()
142 Tensor reordered = mobile::empty_with_tail_padding( in reorder_weights_for_transpose_conv()304 output = mobile::empty_with_tail_padding( in run()316 output = mobile::empty_with_tail_padding( in run()
23 Tensor output = mobile::empty_with_tail_padding( in global_average_pool()
48 Tensor output_padded_contig_nhwc = mobile::empty_with_tail_padding( in channel_shuffle()
134 Tensor output = mobile::empty_with_tail_padding( in run()
167 Tensor output_padded_contig_nhwc = mobile::empty_with_tail_padding( in max_pool2d()
1103 # This is moved to aten_cpu because some of the custom ops use empty_with_tail_padding1105 # empty_with_tail_padding into an op and use dispatcher with it. But exposing it as an op