| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/base/ |
| D | crop_base.cc | 37 size_t input_dim = input_shape_.size(); in ReSize() local
46 void CropBaseCPUKernel::PadOffset(int input_dim, CropParameter *crop_para) const { in PadOffset()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | l2_normalize_info.cc | 38 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy() local
69 size_t input_dim = inputs_shape_.at(0).size(); in GenerateOpStrategies() local
|
| D | loss_info.cc | 46 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy() local
137 size_t input_dim = inputs_shape_[0].size(); in GenerateOpStrategies() local
|
| D | gatherd_info.cc | 43 int64_t input_dim = SizeToLong(inputs_shape_[0].size()); in GetAttrs() local
|
| D | activation_info.cc | 116 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy() local
193 size_t input_dim = inputs_shape_.at(0).size(); in GenerateOpStrategies() local
|
| D | reduce_method_info.cc | 52 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
420 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/ |
| D | leaky_relu_int8.cc | 81 auto input_dim = input_tensor->shape().size(); in ReSize() local
|
| /third_party/mindspore/tests/st/export_and_load/ |
| D | test_bgcf.py | 96 input_dim): argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/ |
| D | random_choice_with_mask_cpu_kernel.cc | 132 auto *input_dim = reinterpret_cast<int *>(workspace[0]->addr); in Launch() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/auto_parallel/rec_core/ |
| D | rec_generate_strategy.cc | 336 size_t input_dim = ops[iter_ops]->inputs_tensor_info()[0].shape().size(); in PrepareL2Normalize() local
385 int64_t input_dim = SizeToLong(ops[iter_ops]->inputs_tensor_info()[0].shape().size()); in PrepareAxisRelatedStrategy() local
825 auto input_dim = ops[iter_ops]->inputs_tensor_info()[0].shape().size(); in GetDimList() local
875 auto input_dim = ops[iter_ops]->inputs_tensor_info()[0].shape().size(); in GetDimListFromAttrs() local
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | wide_and_deep.py | 81 def __init__(self, input_dim, output_dim, weight_bias_init, act_str, argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16/ |
| D | crop_fp16.c | 24 int input_dim = para->input_dim_; in Fp16Crop() local
|
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | deepfm.py | 131 …def __init__(self, input_dim, output_dim, weight_bias_init, act_str, scale_coef=1.0, convert_dtype… argument
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
| D | crop_int8.c | 22 int input_dim = para->input_dim_; in Int8Crop() local
|
| /third_party/mindspore/third_party/proto/caffe/ |
| D | caffe.proto | 75 repeated int32 input_dim = 4; field
746 optional uint32 input_dim = 2; field
|