Searched defs:oT (Results 1 – 5 of 5) sorted by relevance
118 int64_t oT = output_size[0]; in max_unpooling3d_shape_check() local178 int64_t oT = output_size[0]; in max_unpooling3d_forward_out_cpu() local
57 const int64_t oT, in max_unpooling3d_forward_kernel()97 int64_t oT, in max_unpooling3d_backward_kernel()220 int64_t oT = output_size[0]; in max_unpooling3d_shape_check() local305 int64_t oT = output_size[0]; in max_unpooling3d_forward_out_cuda() local508 int64_t oT = output_size[0]; in max_unpooling3d_backward_out_cuda() local
41 const int oT = output.size(2); in conv_depthwise3d_cuda_kernel() local112 const int oT = grad_output.size(2); in conv_depthwise3d_cuda_backward_input_kernel() local203 const int oT = grad_output.size(2); in conv_depthwise3d_cuda_backward_weight_kernel() local
109 int64_t oT, in spatial_dilated_max_pooling3d()401 int64_t oT = pooling_output_shape(iT, kT, pT, sT, dT, ceil_mode); in q_maxpool_3d() local
1585 int64_t oT, in qmaxpool_3d_nthwc_kernel()