| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | concat_int8.cc | 34 auto input_num = in_tensors_.size(); in Init() local
75 auto input_num = in_tensors_.size(); in ReSize() local
121 auto input_num = concat_param_->input_num_; in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | pack.cu | 22 __global__ void Pack(const size_t size, const size_t input_num, const size_t dims_behind_axis, T** … in Pack()
33 void PackKernel(const size_t size, const size_t input_num, in PackKernel()
|
| D | concatv2_impl.cu | 22 __global__ void Concat(const size_t size, const int input_num, const int all_size_before_axis, cons… in Concat()
47 void ConcatKernel(const size_t size, const int input_num, const int all_size_before_axis, const int… in ConcatKernel()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | mul.cc | 33 const int64_t input_num = 2; in InferShape() local
46 const int64_t input_num = 2; in InferType() local
|
| D | sub.cc | 33 const int64_t input_num = 2; in InferShape() local
46 const int64_t input_num = 2; in InferType() local
|
| D | soft_shrink.cc | 33 const int64_t input_num = 1; in InferShape() local
44 const int64_t input_num = 1; in InferType() local
|
| D | real_div.cc | 32 const int64_t input_num = 2; in InferShape() local
46 const int64_t input_num = 2; in InferType() local
|
| D | erfinv.cc | 30 const int64_t input_num = 1; in ErfinvInferShape() local
44 const int64_t input_num = 1; in ErfinvInferType() local
|
| D | masked_fill.cc | 30 const int64_t input_num = 3; in InferShape() local
56 const int64_t input_num = 3; in InferType() local
|
| D | lerp.cc | 31 const int64_t input_num = 3; in InferShape() local
56 const int64_t input_num = 3; in InferType() local
|
| D | index_add.cc | 28 const int64_t input_num = 3; in IndexAddInferShape() local
61 const int64_t input_num = 3; in IndexAddInferType() local
|
| D | apply_adagrad_d_a.cc | 32 const int64_t input_num = 8; in InferShape() local
62 const int64_t input_num = 8; in InferType() local
|
| D | sparse_to_dense.cc | 31 const int64_t input_num = 3; in SparseToDenseInfer() local
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | hshrink_grad.cc | 31 const int64_t input_num = 2; in HShrinkGradInferShape() local
45 const int64_t input_num = 2; in HShrinkGradInferType() local
|
| D | hsigmoid_grad.cc | 35 const int64_t input_num = 2; in InferShape() local
50 const int64_t input_num = 2; in InferType() local
|
| D | avg_pool_3d_grad.cc | 33 const int64_t input_num = 2; in InferShape() local
52 const int64_t input_num = 2; in InferType() local
|
| D | relu_grad.cc | 33 const int64_t input_num = 2; in InferShape() local
51 const int64_t input_num = 2; in InferType() local
|
| /third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
| D | concat_int8_coder.cc | 27 int MallocQuantArgForConcat(ConcatQuantArg *quant_arg, size_t input_num) { in MallocQuantArgForConcat()
40 size_t input_num = input_tensors().size(); in Prepare() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_simple_mem_plan.cc | 28 size_t input_num = AnfAlgo::GetInputTensorNum(kernel); in MemPlan() local
72 size_t input_num = AnfAlgo::GetInputTensorNum(kernel); in MemAssign() local
|
| D | kernel_select_cpu.cc | 81 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in GetInputDtypes() local
95 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in GetInputFormat() local
151 auto input_num = input_types.size(); in GetInputDtypeFormatMatchedNum() local
166 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in ExpandKernelAttr() local
195 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in KernelNotSupportException() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | print_reduce_fusion.cc | 36 size_t input_num = AnfAlgo::GetInputTensorNum(node); in GenerateKernelBuildInfo() local
69 size_t input_num = AnfAlgo::GetInputTensorNum(node); in GetOptList() local
136 size_t input_num = AnfAlgo::GetInputTensorNum(cnode); in Run() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 42 bool SetIOIputSize(const std::shared_ptr<AnfNode> &anf_node, const size_t &input_num, in SetIOIputSize()
82 size_t input_num = AnfAlgo::GetInputTensorNum(anf_node); in SetIOSize() local
182 size_t input_num = AnfAlgo::GetInputTensorNum(anf_node); in SetNodeInputs() local
295 const std::shared_ptr<AnfNode> &anf_node, size_t input_num) { in SetExtInfoInputShapeType()
375 size_t input_num = AnfAlgo::GetInputTensorNum(anf_node); in CreateExtInfo() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | convert_cast_format.cc | 45 size_t input_num = AnfAlgo::GetInputTensorNum(cnode); in Process() local
122 size_t input_num = AnfAlgo::GetInputTensorNum(cast_out_node); in CalculateFormat() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/base/ |
| D | stack_base.c | 18 void Stack(void **inputs, void *output, size_t input_num, size_t copy_size, int outer_start, int ou… in Stack()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | kernel_select_graph_kernel.cc | 62 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in ResetKernelBuildInfo() local
200 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in GetDefaultFormat() local
252 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in UpdateInputsKernelInfo() local
353 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in CheckFormatsAndDtypes() local
401 size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node); in UpdateFormatsAndDtypes() local
|