Searched defs:input_strides (Results 1 – 13 of 13) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | sparse_reshape.cc | 35 … const int64_t *input_strides, const int64_t *output_strides, in SpecialCompute()
142 int64_t *input_strides = new int64_t[input_rank]; in Compute() local
|
| D | densetodense_set_operation.cc | 121 …mDenseGroup(CpuKernelContext &ctx, Tensor *input_tensor, const std::vector<int64_t> &input_strides, in PopulateFromDenseGroup()
|
| D | densetosparsesetoperation.cc | 109 … const std::vector<int64_t> &input_strides, in PopulateFromDenseGroup()
|
| D | pad_v3.cc | 423 std::vector<int64_t> input_strides(input_dims, 0); in ConstantModeCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | mirror_pad_grad_cpu_kernel.cc | 130 std::vector<int64_t> input_strides, std::vector<T> inputs, in slice()
184 std::vector<int64_t> input_strides(dims_, 0); in LaunchKernel() local
|
| D | spacetodepth_cpu_kernel.cc | 87 int64_t input_strides[3] = {1, 1, 1}; in LaunchKernel() local
|
| D | sparse_reshape_cpu_kernel.cc | 145 std::vector<int64_t> input_strides(input_rank); in LaunchKernel() local
|
| D | dense_to_dense_set_operation_cpu_kernel.cc | 116 …upSet(kernel::KernelTensor *input, const size_t last_dim, const std::vector<size_t> &input_strides, in GetGroupSet()
|
| D | pad_v3_cpu_kernel.cc | 139 std::vector<int64_t> input_strides(input_dim_, 0); in ConstantModeCompute() local
|
| D | dense_to_sparse_set_operation_cpu_kernel.cc | 121 …teFromDenseGroup(kernel::KernelTensor *input, int64_t last_dim, std::vector<int64_t> input_strides, in PopulateFromDenseGroup()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | masked_select_impl.cu | 54 …UnaryBroadcastStrideInfo input_strides = UnaryBroadcastCalStride(dim_size, input_shape, broadcast_… in MaskedSelect() local
|
| D | diagonal_impl.cu | 26 int64_t input_strides[DIM_SIZE]; member
|
| D | masked_select_grad_impl.cu | 68 …UnaryBroadcastStrideInfo input_strides = UnaryBroadcastCalStride(dim_size, input_shape, broadcast_… in MaskedSelectGrad() local
|