| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | l2_normalize_info.cc | 39 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy() local
70 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
116 size_t input_dim = inputs_shape_[0].size(); in GenerateOpStrategies() local
|
| D | elementary_function_info.cc | 42 size_t input_dim = inputs_shape_.at(0).size(); in CheckStrategy() local
|
| D | uniform_candidate_sampler_info.cc | 186 const auto &input_dim = strategy_->GetInputDim(); in replace_graph() local
199 const auto &input_dim = strategy_->GetInputDim(); in ComputeReplaceGraph() local
|
| D | reduce_method_info.cc | 54 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
327 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
433 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
503 auto input_dim = inputs_shape_.at(0).size(); in reduce_dim() local
|
| D | slice_info.cc | 189 const auto &input_dim = strategy_->GetInputDim(); in replace_graph() local
205 const auto input_dim = strategy_->GetInputDim(); in ComputeReplaceGraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | index_fill.cc | 80 void IndexFillCpuKernel::SpecialCompute(int64_t start, int64_t end, const int32_t input_dim, in SpecialCompute()
127 … const int32_t input_dim, std::map<int32_t, bool> &index_dict) { in SpecialComputeParallel()
151 int32_t input_dim; in DoCompute() local
|
| D | pad_v3_grad.h | 57 int64_t input_dim; variable
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/base/ |
| D | crop_base.c | 20 int CropPadOffset(int input_dim, CropParameter *crop_para, int64_t *in_offset) { in CropPadOffset()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | median_grad_impl.cu | 45 … int *elem_num_each_dim_y, int64_t axis, int64_t input_dim, int64_t size) { in MedianGradComputer()
65 … bool global_median, const int64_t input0_size, const int64_t input1_size, int64_t input_dim, in MedianGrad()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | sinh.cc | 26 const auto input_dim = input_shape->GetShapeVector().size(); in InferShape() local
|
| D | layer_norm_ext.cc | 62 const auto input_dim = x_shape.size(); in InferShape() local
133 const auto input_dim = x_shape.size(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/nnacl/int8/ |
| D | leaky_relu_int8_coder.cc | 42 auto input_dim = input_tensor->shape().size(); in Prepare() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | pdist_grad_proto.cc | 34 std::vector<int64_t> input_dim = input_shape.GetDims(); in IMPLEMT_COMMON_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/tests/st/mindscience/mindsponge/mindsponge/cell/ |
| D | transition.py | 34 def __init__(self, num_intermediate_factor, input_dim, batch_size=None, slice_num=0): argument
|
| D | basic.py | 180 def __init__(self, num_head, gating, input_dim, output_dim, batch_size=None): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/ |
| D | leaky_relu_int8.cc | 80 auto input_dim = input_tensor->shape().size(); in ReSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/nn/layer/ |
| D | dense.py | 50 def check_last_dimension(input_dim, input_channels, input_name, input_channels_name, prim_name=None… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | max_pool_grad_grad_with_argmax.cc | 42 const int64_t input_dim = 4; in MaxPoolGradGradWithArgmaxInferShape() local
|
| D | max_pool_grad_grad.cc | 82 const int64_t input_dim = 4; in MaxPoolGradGradInferShape() local
|
| D | max_pool_3d_grad_grad.cc | 55 const int64_t input_dim = 5; in MaxPool3DGradGradInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/nn/ |
| D | lamb_gpu_kernel.h | 137 size_t input_dim = variable_int64_shape.size(); in Resize() local
204 …heckValidShape(const ShapeVector &input_shape, const ShapeVector &output_shape, size_t input_dim) { in CheckValidShape()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | topk_tensorrt.cc | 196 auto input_dim = input(ctx, 0).trt_tensor_->getDimensions(); in PreprocessInputs() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | quantile_cpu_kernel.cc | 145 std::vector<int64_t> SetQuantileOutputShape(int64_t dim, int64_t input_dim, bool keep_dims, const i… in SetQuantileOutputShape()
179 size_t input_dim = input_shape_.size(); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | pdist.cc | 54 const int64_t input_dim = 2; in PdistInferShape() local
|