| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/ |
| D | sparse_split_impl.cu | 26 … IndexType *shape_ptr, IndexType num_split, IndexType **y_indices_ptr, in SparseSplitKernel()
56 … IndexType *shape_ptr, IndexType num_split, IndexType **y_indices_ptr, in SparseSplit()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | meshgrid.cc | 59 …auto shape_ptr = std::make_shared<abstract::Shape>(std::vector<int64_t>{abstract::Shape::kShapeRan… in MeshgridInferShape() local
72 auto shape_ptr = std::make_shared<abstract::Shape>(output_shape); in MeshgridInferShape() local
|
| D | tensor_to_scalar.cc | 52 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape(op_name, input_args, 0); in InferShape() local
|
| D | map_tensor_get_values.cc | 42 auto shape_ptr = abs_map_tensor->GetShape(); in MapTensorGetValuesInferShape() local
|
| D | arg_min_v2.cc | 43 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape("ArgminV2", input_args, 0); in ArgminV2InferShape() local
|
| D | randperm.cc | 102 auto shape_ptr = input_args[kInputIndex0]->GetShape(); in InferShape() local
|
| D | apply_ftrl.cc | 105 auto shape_ptr = var_shape->cast<abstract::ShapePtr>(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | dropout.cc | 31 auto shape_ptr = std::make_shared<abstract::Shape>(x_shape); in InferShape() local
|
| D | range.cc | 93 BaseShapePtr shape_ptr = nullptr; in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fisson/ |
| D | node_out_shapes.cc | 60 const auto &shape_ptr = (*shape)[i]; in Run() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/ |
| D | reshape_checker.cc | 56 auto shape_ptr = primitive->GetAttr(ops::kShape); in Check() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | sparse_count_sparse_output_cpu_kernel.cc | 44 …arseOutputCpuKernelMod::CheckIndicesInBounds(const int64_t *indices_addr, const int64_t *shape_ptr, in CheckIndicesInBounds()
153 const auto *shape_ptr = reinterpret_cast<int64_t *>(inputs[2]->device_ptr()); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/environ/ |
| D | aicpu_environ_manager.cc | 62 auto shape_ptr = tensor->GetTensorShape(); in IsScalarTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/ |
| D | storage_format_operator.cc | 33 auto shape_ptr = base_shape->cast<abstract::ShapePtr>(); in GetShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | apply_momentum_scale_fusion.cc | 33 auto shape_ptr = in->Shape(); in IsScalar() local
|
| D | apply_momentum_weight_scale_fusion.cc | 41 auto shape_ptr = in->Shape(); in IsScalar() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/ |
| D | utils.cc | 103 bool ParseShapeStr(const std::string &shape_str, std::vector<int64_t> *shape_ptr) { in ParseShapeStr()
130 bool ParseShapeStr(const std::string &shape_str, std::vector<ShapeDim> *shape_ptr) { in ParseShapeStr()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fission/ |
| D | transdata_split.cc | 45 auto shape_ptr = base_shape->cast<abstract::ShapePtr>(); in IsDepthwiseCase() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/ |
| D | basic_op_infer_shape.cc | 63 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape(primitive->name(), input_args, 0); in AllGatherInferShape() local
84 auto shape_ptr = CheckAndConvertUtils::GetTensorInputShape(primitive->name(), input_args, 0); in ShapeInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/parser/ |
| D | parser_utils.cc | 137 auto shape_ptr = api::MakeShared<api::Shape>(shape); in UnifyVariableConvWeight() local
201 auto shape_ptr = api::MakeShared<api::Shape>(shape); in HandleConstConvWeightShared() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/cxx_api/model/ms/ |
| D | ms_model.cc | 63 auto shape_ptr = std::dynamic_pointer_cast<abstract::Shape>(param->Shape()); in GenerateGraphCell() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/convert/ |
| D | runtime_convert.cc | 93 auto shape_ptr = std::make_shared<mindspore::abstract::Shape>(shape_vec); in UpdateDynamicInputShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sparse/ |
| D | sparse_split_gpu_kernel.cc | 84 auto shape_ptr = GetDeviceAddress<IndexType>(inputs, kIndex3); in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/ |
| D | parser_utils.cc | 310 auto shape_ptr = std::make_shared<abstract::Shape>(shape); in UnifyVariableConvWeight() local
411 auto shape_ptr = std::make_shared<abstract::Shape>(shape); in HandleConstConvWeightShared() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/format_type/ |
| D | eliminate_graph_output_transdata.cc | 43 auto shape_ptr = base_shape->cast<abstract::ShapePtr>(); in IsDepthwiseCase() local
|