| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/control/ |
| D | tensorlist_reserve_infer.c | 62 vvector tmp_shape; in TensorListReserveInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/expanders/ |
| D | minimum_grad.cc | 37 auto tmp_shape = ShapeVector(brodcast_shape.size() - original_shape.size(), 1); in GetReduceAxis() local
|
| D | maximum_grad.cc | 37 auto tmp_shape = ShapeVector(brodcast_shape.size() - original_shape.size(), 1); in GetReduceAxis() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/micro/coder/opcoders/base/ |
| D | reduce_base_coder.cc | 97 std::vector<int> tmp_shape = input_tensors_.at(0)->shape(); in CalculateInnerOuterSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | spacetobatch_gpu_kernel.cc | 77 …auto tmp_shape = input_shape_[idx_i + PADDING_SHAPE_1] + paddings_[idx_i][0] + paddings_[idx_i][1]; in Resize() local
|
| D | batchtospace_gpu_kernel.h | 122 …auto tmp_shape = input_shape[idx_i + CROPS_SHAPE_1] * block_size_ - crops_[idx_i][0] - crops_[idx_… in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/base/ |
| D | reduce_base.cc | 136 auto tmp_shape = in_tensors_.at(0)->shape(); in CalculateInnerOuterSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | space_to_batch_nd_cpu_kernel.cc | 60 auto tmp_shape = input_shape_[idx_i + off_set_] + paddings_[idx_i][0] + paddings_[idx_i][1]; in CheckParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_class/ |
| D | space_to_batch_nd_helper.h | 152 auto tmp_shape = input_shape_[idx_i + off_set] + paddings_[idx_i][0] + paddings_[idx_i][1]; in CheckKernelParam() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | space_to_batch_nd.cc | 92 auto tmp_shape = input_shape_[idx_i + off_set_] + paddings_[idx_i][0] + paddings_[idx_i][1]; in SpaceToBatchNDInferShapeImpl() local
|
| D | fill.cc | 209 auto tmp_shape = InferShape(prim, input_args); in InferValue() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/ |
| D | third_party_param_parser.cc | 54 for (auto tmp_shape : tmp_shapes) { in DoParseShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/engine/ |
| D | data_schema.cc | 281 std::vector<dsize_t> tmp_shape = {}; in ColumnLoad() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/grad/ |
| D | conv2d_backprop_input.cc | 122 ShapeVector tmp_shape = {dout_shape[0], dout_shape[2], dout_shape[3], dout_shape[1]}; in Conv2DBackpropInputInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | weight_quant_batch_matmul_info.cc | 564 Shape tmp_shape = GetCommonShape(mat_x_shape, mat_weight_shape); in GenerateOpStrategies() local
|
| D | quant_batch_matmul_info.cc | 563 Shape tmp_shape = GetCommonShape(x1_shape, x2_shape); in GenerateOpStrategies() local
|
| D | conv2d_info.cc | 1046 Shape tmp_shape = inputs_shape_[0]; in GenerateOpStrategies() local
|
| D | matmul_info.cc | 996 Shape tmp_shape = GetCommonShape(mat_a_shape, mat_b_shape); in GenerateOpStrategies() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/abstract/ops/ |
| D | prim_others.cc | 245 auto tmp_shape = x->shape()->shape(); in InferImplReduceScatter() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/ |
| D | decrease_transpose_algo.cc | 235 ShapeVector tmp_shape(DIMENSION_4D - expand_shape.size(), 1); in ConvertTensorToNCOrNH() local
|