Searched defs:offsetZ (Results 1 – 11 of 11) sorted by relevance
43 int offsetZ, int divisor_override) in avg_pool3d_cuda_update_output()112 int offsetZ, int divisor_override) in avg_pool3d_cuda_update_output()176 int offsetZ) in avg_pool3d_single_backward_out_frame_stride1()223 int offsetZ, int divisor_override, const int gradInput_numel) in avg_pool3d_cuda_update_grad_input_atomic()281 bool count_include_pad, int offsetZ, int divisor_override) in avg_pool3d_cuda_update_grad_input()406 int64_t offsetZ = 0; in TORCH_IMPL_FUNC() local529 int64_t offsetZ = 0; in TORCH_IMPL_FUNC() local565 int64_t offsetZ = 0; in TORCH_IMPL_FUNC() local
52 int64_t offsetZ) in adaptivemaxpool()133 int64_t offsetZ = 0; in adaptivemaxpool_loop() local164 int64_t offsetZ in adaptivemaxgradinput()211 int64_t offsetZ = 0; in adaptivemaxgradinput_loop() local240 int64_t offsetZ in atomicadaptivemaxgradinput()287 int64_t offsetZ = 0; in atomicadaptivemaxgradinput_loop() local
58 int64_t offsetZ) { in adaptiveaveragepool()125 int64_t offsetZ = 0; in adaptiveaveragepool_loop() local157 int64_t offsetZ) in adaptiveaveragegradinput()222 int64_t offsetZ = 0; in adaptiveaveragegradinput_loop() local255 int64_t offsetZ) in atomicadaptiveaveragegradinput()318 int64_t offsetZ = 0; in atomicadaptiveaveragegradinput_loop() local
60 const int64_t offsetZ) { in max_unpooling3d_forward_kernel()102 int offsetZ) { in max_unpooling3d_backward_kernel()358 int offsetZ = 0; in max_unpooling3d_forward_out_cuda() local568 int offsetZ = 0; in max_unpooling3d_backward_out_cuda() local
45 int offsetZ, in max_pool3d_with_indices_single_out_frame()146 int offsetZ = 0; in max_pool3d_with_indices_out_frame() local193 int offsetZ, in max_pool3d_with_indices_backward_single_out_frame()249 int offsetZ = 0; in max_pool3d_with_indices_backward_out_frame() local
928 … for (uint32_t offsetZ = offset.depth; offsetZ < offset.depth + extent.depth; in iterate() local1015 … for (uint32_t offsetZ = offset.depth; offsetZ < offset.depth + extent.depth; in iterate() local1076 … for (uint32_t offsetZ = 0u; offsetZ < planeExtent.depth * imageCreateInfo.arrayLayers; ++offsetZ) in iterate() local
782 for (uint32_t offsetZ = 0u; offsetZ < gridSize.z(); ++offsetZ) in iterate() local
485 for (uint32_t offsetZ = 0u; offsetZ < planeExtent.depth; ++offsetZ) in testStorageImageWrite() local
884 offsetZ; /* Offsets from the start of the GlyphVariationData array member
3025 uint32_t offsetZ = baseGroupZ + localIdxZ * m_localSize.z(); in iterate() local
3415 GLint offsetZ, in ValidateCopyImageSubDataTargetRegion()