| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | triu.cc | 25 auto input_shape_vec = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in InferShape() local
|
| D | gather_d_grad_v2.cc | 34 auto input_shape_vec = input_args[kInputIndex0]->GetShape()->GetShapeVector(); in CheckValidation() local
|
| D | rank.cc | 42 auto input_shape_vec = x_abs->GetShape()->GetShapeVector(); in InferValue() local
|
| D | scatter_add_ext.cc | 39 auto input_shape_vec = input_shape->GetShapeVector(); in InferShape() local
|
| D | eltwise_op.cc | 35 auto input_shape_vec = input_shape->GetShapeVector(); in InferShapeWithCheck() local
|
| D | reshape.cc | 28 auto input_shape_vec = input_shape->GetShapeVector(); in InferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ |
| D | unique_consecutive.cc | 51 auto input_shape_vec = input_shape_map[kShape]; in UniqueConsecutiveInferShape() local
112 auto input_shape_vec = input_shape_map[kShape]; in UniqueConsecutiveFrontendInferShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/delegate/tensorrt/op/ |
| D | slicefusion_tensorrt.cc | 33 std::vector<int> input_shape_vec; in GetDynamicSliceSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/utils/ |
| D | reduce_infer_util.cc | 138 std::vector<int64_t> input_shape_vec = tensordesc_input_x.GetShape().GetDims(); in DoReduceInferRangeWithAxes() local
213 std::vector<int64_t> input_shape_vec = input_shape.GetDims(); in DoReduceInferShapeWithoutAxes() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/ |
| D | masked_fill_gpu_kernel.cc | 124 std::vector<int64_t> input_shape_vec = inputs.at(kIndex0)->GetShapeVector(); in Resize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/ |
| D | micro_param_parser.cc | 132 auto input_shape_vec = SplitStringToVector(input_shape, ','); in ParseGraphInputsShapeTemplate() local
|
| D | config_file_parser.cc | 230 auto input_shape_vec = SplitStringToVector(input_shape, ','); in ParseInputShapeTemplate() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/ |
| D | max_pool3d_with_argmax_cpu_kernel.cc | 193 auto input_shape_vec = x_shape_; in LaunchKernel() local
|
| D | max_pool3d_grad_with_argmax_cpu_kernel.cc | 164 auto input_shape_vec = x_shape_; in LaunchKernel() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/ |
| D | max_pool_3d_grad_with_argmax.cc | 231 auto input_shape_vec = input_x_info->GetTensorShape()->GetDimSizes(); in MaxPool3DGradWithArgmaxCompute() local
|
| D | max_pool_3d_with_argmax.cc | 258 auto input_shape_vec = input_info->GetTensorShape()->GetDimSizes(); in MaxPool3DWithArgmaxCompute() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/ |
| D | audio_utils.cc | 2488 auto input_shape_vec = input->shape().AsVector(); in MelSpectrogram() local
|