| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | shape_tensorrt.cc | 39 nvinfer1::ITensor *shape_input = input(ctx, 0).trt_tensor_; in AddInnerOp() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/fusion/ |
| D | reshape_shape_fusion.cc | 76 auto shape_input = shape->input(1); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/acl/mapper/ |
| D | reshape_mapper.cc | 61 auto shape_input = cnode->input(kInputNum - 1); in ShapeAttrAdjustTensor() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/ascend_native/ |
| D | ascend_native_gather_kernel.cc | 28 auto shape_input = in_tensors_.at(FIRST_INPUT)->shape(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | diag_part.cc | 60 std::vector<int64_t> shape_input = ctx.Input(0)->GetTensorShape()->GetDimSizes(); in DiagPartCheck() local
|
| D | cumprod.cc | 86 std::vector<int64_t> shape_input = ctx.Input(0)->GetTensorShape()->GetDimSizes(); in CumprodCheck() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | truncated_normal_cpu_kernel.cc | 66 auto shape_input = inputs[kIndex0]->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | expand.cc | 105 auto shape_input = input_args[1]; in ExpandInferShape() local
|
| D | truncated_normal.cc | 61 auto shape_input = shape_input_map[kShape]; in TruncatedNormalInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/ |
| D | random_ops_proto.cc | 72 ge::Shape shape_input = op.GetInputDescByName("input").GetShape(); in IMPLEMT_COMMON_INFERFUNC() local
115 ge::Shape shape_input = op.GetInputDescByName("x").GetShape(); in IMPLEMT_COMMON_INFERFUNC() local
|
| D | selection_ops_proto.cc | 65 ge::Shape shape_input = op.GetInputDescByName("x").GetShape(); in IMPLEMT_COMMON_INFERFUNC() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | graph_utils.cc | 303 std::vector<AnfNodePtr> *shape_input, in MatchingAccordingToPrime()
354 std::vector<AnfNodePtr> *shape_input, in MatchingAccordingToIndex()
403 int64_t CountDynamicAxis(const AnfNodePtrList &shape_input) { in CountDynamicAxis()
470 std::vector<AnfNodePtr> shape_input; in ConvertConstParamToDynamic() local
1304 auto shape_input = reshape_node->input(INDEX_TWO); in SegmentEntireShapeToPartialForDynamic() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/ |
| D | shape_reshape_fusion.cc | 268 auto shape_input = shape_cnode->input(kIndex1); in Process() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/optimizer/ |
| D | tensorrt_optimizer.cc | 105 AnfNodePtr *shape_input) { in GetMatmulFactor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_redistribution.cc | 650 …auto shape_input = UpdateShapeNodeInput(cur_cnode, pre_cnode->cast<CNodePtr>(), redistribution_ind… in CreateAssembledDynamicMapping() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/_extends/parse/ |
| D | standard_method.py | 2078 def _check_sum_to_size(size, input_dim, shape_input): argument
2085 def _count_axes(size, input_shape, shape_input): argument
|