Home
last modified time | relevance | path

Searched defs:input_stride (Results 1 – 25 of 74) sorted by relevance

123

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dconjugate_transpose_impl.cu27 __global__ void ConjugateTranspose(const size_t size, const T *input, const size_t *input_stride, in ConjugateTranspose()
52 …bal__ void ConjugateTransposeComplex(const size_t size, const T *input, const size_t *input_stride, in ConjugateTransposeComplex()
77 cudaError_t CalConjugateTranspose(const size_t size, const T *input, const size_t *input_stride, in CalConjugateTranspose()
86 …Error_t CalConjugateTransposeComplex(const size_t size, const T *input, const size_t *input_stride, in CalConjugateTransposeComplex()
Dmaxpool_grad_grad_with_argmax_impl.cu22 __global__ void MaxPoolGradGradWithArgmax(const T *grad, const I *argmax, const int input_stride, in MaxPoolGradGradWithArgmax()
31 …lMaxPoolGradGradWithArgmax(const T *grad, const I *argmax, const int batch, const int input_stride, in CalMaxPoolGradGradWithArgmax()
Dbroadcast_to_impl.cuh23 size_t input_stride[8]; member
Dgrid_sampler_impl.cu138 … const std::vector<size_t> &output_shape, const std::vector<size_t> &input_stride, in GridSampler2D()
307 … const std::vector<size_t> &output_shape, const std::vector<size_t> &input_stride, in GridSampler3D()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/
Dsplit_with_over_lap_base.c27 int input_stride = param->split_dim_size_ * param->inner_stride_ * param->element_bytes_; in DoSplitWithOverlapParallel() local
Dconcat_base.c38 int input_stride = after_axis_size * inputs_output_shape[i][axis]; in Concat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dconjugate_transpose_gpu_kernel.h76 size_t input_stride[7]; variable
/third_party/mesa3d/src/gallium/auxiliary/draw/
Ddraw_vs_llvm.c58 unsigned input_stride, in vs_llvm_run_linear()
Ddraw_vs_exec.c97 unsigned input_stride, in vs_exec_run_linear()
Ddraw_tess.c166 unsigned input_stride = input_verts->vertex_size; in draw_tess_ctrl_shader_run() local
333 unsigned input_stride = input_verts->vertex_size; in draw_tess_eval_shader_run() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/fp32/
Dconv_im2col_fp32.c45 int input_stride = (input_h * in_w + input_w) * in_channel; in Im2ColDataPackUnitFp32() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/intrinsics/sse/
DConvDwFp32IndirectRow.c28 size_t output_width, size_t input_stride, size_t relu, size_t relu6) { in ConvDwFp32Avx5x5()
/third_party/vixl/test/aarch32/config/
Dtemplate-simulator-aarch32.cc.in192 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; local
/third_party/vixl/test/aarch32/
Dtest-simulator-rd-rn-rm-a32.cc491 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rd-rn-operand-imm12-t32.cc924 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-rd-rn-rm-t32.cc491 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rdlow-operand-imm8-t32.cc1589 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rd-operand-imm16-t32.cc437 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rd-operand-const-t32.cc599 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rd-operand-const-a32.cc484 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-dt-drt-drd-drn-drm-float-f64-a32.cc1411 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rdlow-rnlow-operand-immediate-t32.cc970 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
Dtest-simulator-cond-rd-rn-a32.cc869 size_t input_stride = sizeof(kTests[i].inputs[0]) * kTests[i].input_size; in TestHelper() local
/third_party/mesa3d/src/amd/vulkan/
Dradv_query.c208 nir_def *input_stride = nir_imm_int(&b, db_count * 16); in build_occlusion_query_shader() local
496 nir_def *input_stride = in build_pipeline_statistics_query_shader() local
889 nir_def *input_stride = nir_imm_int(&b, 32); in build_tfb_query_shader() local
1151 nir_def *input_stride = nir_imm_int(&b, 8); in build_timestamp_query_shader() local
1297 nir_def *input_stride = in build_pg_query_shader() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dunique_consecutive_cpu_kernel.cc102 std::vector<int64_t> input_stride = ConstructStride<int64_t>(input_shape_); in ReshapeInput() local

123