| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | sequence_slice_grad.cc | 42 constexpr size_t input_num = 5; in SliceGradInferInner() local
56 constexpr size_t input_num = 5; in InferShape() local
65 constexpr size_t input_num = 5; in InferType() local
|
| D | sequence_slice_setitem.cc | 35 constexpr size_t input_num = 5; in SequenceSliceInferInner() local
93 constexpr size_t input_num = 5; in InferShape() local
130 constexpr size_t input_num = 5; in InferTypeInner() local
166 constexpr size_t input_num = 5; in InferType() local
|
| D | unravel_index.cc | 49 const int64_t input_num = 2; in UravelIndexInferShape() local
88 const int64_t input_num = 2; in UravelIndexInferType() local
99 const int64_t input_num = 2; in UnravelIndexInfer() local
|
| D | shape_mul.cc | 45 const int64_t input_num = 1; in InferShape() local
61 const int64_t input_num = 1; in InferValue() local
|
| D | send.cc | 62 const int64_t input_num = 1; in InferType() local
75 const int64_t input_num = 1; in InferShapeAndType() local
|
| D | all_reduce.cc | 58 const int64_t input_num = 1; in InferType() local
88 const int64_t input_num = 1; in InferShapeAndType() local
|
| D | all_to_all_ascend.cc | 58 const int64_t input_num = 1; in InferType() local
88 const int64_t input_num = 1; in InferShapeAndType() local
|
| D | l2_loss.cc | 42 const int64_t input_num = 1; in L2LossInferShape() local
50 const int64_t input_num = 1; in L2LossInferType() local
|
| D | non_max_suppression_v3.cc | 46 const int input_num = 5; in NonMaxSuppressionV3InferShape() local
86 const int input_num = 5; in NonMaxSuppressionV3FrontendInferShape() local
128 const int input_num = 5; in NonMaxSuppressionV3InferType() local
|
| D | adam_no_update_param.cc | 49 const int64_t input_num = 9; in InferShape() local
71 const int64_t input_num = 9; in InferType() local
|
| D | truncate_mod.cc | 50 const int64_t input_num = 2; in TruncateModInferShape() local
61 const int64_t input_num = 2; in TruncateModInferType() local
|
| D | l2_normalize.cc | 76 const int64_t input_num = kL2NormalizeInputsNum; in InferShape() local
103 const int64_t input_num = kL2NormalizeInputsNum; in InferType() local
|
| D | sparse_matrix_transpose.cc | 53 const int64_t input_num = 5; in SparseMatrixTransposeInferShape() local
128 const int64_t input_num = 5; in SparseMatrixTransposeInferType() local
167 const int64_t input_num = 5; in SparseMatrixTransposeInfer() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | concat_int8.cc | 40 auto input_num = in_tensors_.size(); in Prepare() local
87 auto input_num = in_tensors_.size(); in ReSize() local
112 auto input_num = in_tensors().size(); in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | parallel_concat_impl.cu | 28 __global__ void ParallelConcat(const size_t size, const int input_num, T **inputs, T *output) { in ParallelConcat()
39 cudaError_t ParallelConcatKernel(const size_t size, const int input_num, T **inputs, T *output, in ParallelConcatKernel()
|
| D | pack.cu | 24 __global__ void Pack(const size_t size, const size_t input_num, const size_t dims_behind_axis, T **… in Pack()
35 cudaError_t PackKernel(const size_t size, const size_t input_num, const size_t dims_behind_axis, T … in PackKernel()
|
| D | concat_impl.cu | 28 __global__ void Concat(const size_t size, const int input_num, const int all_size_before_axis, cons… in Concat()
53 cudaError_t ConcatKernel(const size_t size, const int input_num, const int all_size_before_axis, in ConcatKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/ |
| D | concat_int8_coder.cc | 28 int MallocQuantArgForConcat(ConcatQuantArg *quant_arg, size_t input_num) { in MallocQuantArgForConcat()
43 size_t input_num = input_tensors().size(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | softplus_grad.cc | 46 const int64_t input_num = 2; in SoftplusGradInfershape() local
72 const int64_t input_num = 2; in SoftplusGradInfertype() local
|
| D | l2_normalize_grad.cc | 58 const int64_t input_num = kPReLUGradInputsNum; in InferShape() local
91 const int64_t input_num = kPReLUGradInputsNum; in InferType() local
|
| D | masked_select_grad.cc | 46 const int64_t input_num = 3; in MaskedSelectGradInferShape() local
63 const int64_t input_num = 3; in MaskedSelectGradInferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/st/scripts/experimental/ |
| D | model_info.py | 36 def _input_data_str(input_file_prefix, input_num): argument
49 def init(self, input_num=1, input_suffix="", output_suffix=""): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/providers/nnie/ |
| D | nnie_micro.cc | 104 static int NnieKernel(TensorC *inputs, int input_num, TensorC *outputs, int output_num, CustomParam… in NnieKernel()
142 static int ProposalKernel(TensorC *inputs, int input_num, TensorC *outputs, int output_num, CustomP… in ProposalKernel()
211 int CustomKernel(TensorC *inputs, int input_num, TensorC *outputs, int output_num, CustomParameter … in CustomKernel()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/ |
| D | cpu_simple_mem_plan.cc | 29 size_t input_num = common::AnfAlgo::GetInputTensorNum(kernel); in MemPlan() local
73 size_t input_num = common::AnfAlgo::GetInputTensorNum(kernel); in MemAssign() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | adam_weight_decay_gpu_kernel.cc | 23 constexpr size_t input_num = 10; in Init() local
59 constexpr size_t input_num = 10; in Resize() local
|