Searched defs:output_batch_stride (Results 1 – 5 of 5) sorted by relevance
467 size_t output_batch_stride; member496 size_t output_batch_stride; member532 size_t output_batch_stride; member554 size_t output_batch_stride; member604 size_t output_batch_stride; member638 size_t output_batch_stride; member677 size_t output_batch_stride; member716 size_t output_batch_stride; member746 size_t output_batch_stride; member777 size_t output_batch_stride; member[all …]
103 int64_t output_batch_stride = output.stride(0); in col2im_out_cuda_template() local
1381 def _bsr_strided_dense_rowspace_kernel(2247 def _bsr_strided_addmm_kernel(
582 …const size_t output_batch_stride = (output_height * output_width * convolution_op->output_pixel_st… in setup_convolution2d_nchw() local
532 size_t output_batch_stride; member569 size_t output_batch_stride; member650 size_t output_batch_stride; member