| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | _embedding_cache_ops.py | 52 def check_shape(self, input_x_shape, indices_shape, update_shape, max_num_shape): member in UpdateCache
97 def check_shape(self, input_x_shape, max_num_shape, offset_shape): member in SubAndFilter
172 def check_shape(self, cache_table_shape, swap_cache_idx_shape, miss_value_shape): member in CacheSwapTable
303 def check_shape(self, input_x_shape, cum_sum_arr_shape, shift_idx_shape): member in PadAndShift
|
| D | _inner_ops.py | 550 def check_shape(self, input_shape): member in ConvertToDynamic
600 def check_shape(self, input_shape): member in GpuConvertToDynamicShape
701 def check_shape(self, lengths_shape, maxlen_shape): member in SequenceMask
1023 def check_shape(self, indices_shape, data_shape): member in DynamicStitch
|
| D | array_ops.py | 151 def check_shape(self, x_shape, indices_shape, updates_shape): member in _ScatterOpDynamic
4398 def check_shape(self, x_shape, mask_shape): member in MaskedSelect
|
| D | nn_ops.py | 2754 def check_shape(self, parameters_shape, gradient_shape, learning_rate_shape, member in SGD
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | tensor_scatter_arithmetic.cc | 40 bool CheckShape(const std::vector<int64_t> &updates_shape, const std::vector<int64_t> &check_shape)… in CheckShape()
|
| D | instance_norm_v2.cc | 66 ShapeVector check_shape = x_shape; in InstanceNormV2InputShapeCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | instance_norm_v2_grad.cc | 74 ShapeVector check_shape = dy_shape; in InstanceNormV2GradInputShapeCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/utils/ |
| D | kernel_util.cc | 266 … const std::vector<int64_t> &check_shape, const std::string &prim_name) { in CheckTensorShapeSame()
|
| D | kernel_util.h | 214 …ool ShapeVectorIsSame(const std::vector<int64_t> &shape, const std::vector<int64_t> &check_shape) { in ShapeVectorIsSame()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | im2col_and_col2im_ext.cc | 289 …auto check_shape = Im2ColAndCol2ImCommonCheckShape(primitive, input_args[kIndex0], no_batch_rank_,… in CheckValidation() local
348 …auto check_shape = Im2ColAndCol2ImCommonCheckShape(primitive, input_args[kIndex0], no_batch_rank_,… in CheckValidation() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | instance_norm_v2_grad.cc | 102 std::vector<int64_t> check_shape; in InstanceNormV2GradShapeCheck() local
|
| D | instance_norm_v2.cc | 114 std::vector<int64_t> check_shape; in InstanceNormV2ShapeCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/utils/ |
| D | convert_utils_base.h | 267 …ool ShapeVectorIsSame(const std::vector<int64_t> &shape, const std::vector<int64_t> &check_shape) { in ShapeVectorIsSame()
|
| D | check_convert_utils.cc | 846 const std::vector<int64_t> &check_shape, in CheckTensorShapeSame()
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/ |
| D | primitive.py | 607 def check_shape(self, *args): member in PrimitiveWithCheck
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/debug/ |
| D | debug_services.cc | 867 std::stringstream check_shape(shape_str); in ReadTensorFromNpy() local
|