Home
last modified time | relevance | path

Searched defs:kTwo (Results 1 – 25 of 43) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Dmatrix_triangular_solve.cc68 const int64_t kTwo = 2; in MatrixTriangularSolveInferShape() local
127 const int64_t kTwo = 2; in MatrixTriangularSolveInferType() local
Dorgqr.cc49 const size_t kTwo = 2; in OrgqrInferShape() local
Dangle_atom_energy.cc103 static constexpr int64_t kTwo = 2; member in mindspore::ops::AngleAtomEnergyInfer
Dormqr.cc51 const size_t kTwo = 2; in OrmqrInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Ddense.cc56 const size_t kTwo = 2; in InferShape() local
153 const size_t kTwo = 2; in InferShape() local
Dgeqrf.cc51 const size_t kTwo = 2; in InferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/utils/
Dcommon.h26 constexpr int kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/math/
Dpolar_gpu_kernel.cc25 constexpr size_t kTwo = 2; variable
Dcdist_gpu_kernel.cc24 constexpr size_t kTwo = 2; variable
Dcdist_grad_gpu_kernel.cc24 constexpr size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dlower_bound_cpu_kernel.cc69 const size_t kTwo = 2; in LaunchKernel() local
Dapply_gradient_descent_cpu_kernel.cc26 const size_t kTwo = 2; variable
Dgrid_sampler_2d_grad_cpu_kernel.cc24 const size_t kTwo = 2; variable
Ddense_to_csr_sparse_matrix_cpu_kernel.cc25 constexpr size_t kTwo = 2; variable
Dapply_ada_max_cpu_kernel.cc29 const size_t kTwo = 2; variable
Dcsr_sparse_matrix_to_sparse_tensor_cpu_kernel.cc28 constexpr size_t kTwo = 2; variable
Dsparse_tensor_to_csr_sparse_matrix_cpu_kernel.cc27 constexpr int64_t kTwo = 2; variable
Dmulti_margin_loss_cpu_kernel.cc29 const size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/
Dresize_bilinear_gpu_kernel.cc32 constexpr size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/
Dvsl_compress_tensorrt.cc33 constexpr std::size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/
Dsparse_tensor_to_csr_sparse_matrix_gpu_kernel.cc31 constexpr size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/pipeline/jit/pi/utils/
Dutils.h33 constexpr auto kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/eigen/
Dorgqr_cpu_kernel.cc24 constexpr size_t kTwo = 2; variable
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/
Dpad_v3_grad.cc46 constexpr size_t kTwo = 2; in PadV3GradInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmirror_pad.cc31 constexpr int kTwo = 2; variable

12