Searched refs:patch_stride (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | extract_image_patches_impl.cu | 22 … int64_t patch_stride, int64_t other_stride, int64_t input_row_size, in ExtractImagePatches() argument
31 const int64_t patch_index = inner_index / patch_stride; in ExtractImagePatches()
32 const int64_t patch_offset = (inner_index - patch_index * patch_stride) / output_depth; in ExtractImagePatches()
62 int64_t patch_stride, int64_t other_stride, int64_t input_row_size, in CalExtractImagePatchesNHWC() argument
67 …ize, stride_row, stride_col, rate_row, rate_col, output_cols, need_batch, row_stride, patch_stride, in CalExtractImagePatchesNHWC()
74 … int64_t row_stride, int64_t patch_stride, int64_t other_stride,
81 … bool need_batch, int64_t row_stride, int64_t patch_stride,
89 … int64_t row_stride, int64_t patch_stride, int64_t other_stride,
97 … bool need_batch, int64_t row_stride, int64_t patch_stride,
|
| D | extract_image_patches_impl.cuh | 26 int64_t patch_stride, int64_t other_stride, int64_t input_row_size,
|
| /third_party/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_shader_llvm_tess.c | 114 LLVMValueRef patch_stride = get_tcs_in_patch_stride(ctx); in get_tcs_in_current_patch_offset() local
117 return LLVMBuildMul(ctx->ac.builder, patch_stride, rel_patch_id, ""); in get_tcs_in_current_patch_offset()
123 LLVMValueRef patch_stride = get_tcs_out_patch_stride(ctx); in get_tcs_out_current_patch_offset() local
126 return ac_build_imad(&ctx->ac, patch_stride, rel_patch_id, patch0_offset); in get_tcs_out_current_patch_offset()
132 LLVMValueRef patch_stride = get_tcs_out_patch_stride(ctx); in get_tcs_out_current_patch_data_offset() local
135 return ac_build_imad(&ctx->ac, patch_stride, rel_patch_id, patch0_patch_data_offset); in get_tcs_out_current_patch_data_offset()
|
| /third_party/mesa3d/src/freedreno/ir3/ |
| D | ir3_nir_lower_tess.c | 396 nir_ssa_def *patch_stride = nir_load_hs_patch_stride_ir3(b); in build_per_vertex_offset() local
397 nir_ssa_def *patch_offset = nir_imul24(b, patch_id, patch_stride); in build_per_vertex_offset()
473 const uint32_t patch_stride = 1 + inner_levels + outer_levels; in build_tessfactor_base() local
478 nir_imul24(b, patch_id, nir_imm_int(b, patch_stride)); in build_tessfactor_base()
|