Home
last modified time | relevance | path

Searched defs:output_shape_vec (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/core/ops/
Ddensetodense_set_operation.cc87 ShapeVector output_shape_vec = {output_rank_dim}; in DenseToDenseSetOperationInferShape() local
153 ShapeVector output_shape_vec = {output_rank_dim}; in DenseToDenseSetOperationFrontendInferShape() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/arrays/
Dmasked_fill_gpu_kernel.cc127 std::vector<int64_t> output_shape_vec = outputs.at(kIndex0)->GetShapeVector(); in Resize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dmax_pool3d_with_argmax_cpu_kernel.cc194 auto output_shape_vec = y_shape_; in LaunchKernel() local
Dmax_pool3d_grad_with_argmax_cpu_kernel.cc165 auto output_shape_vec = grads_shape_; in LaunchKernel() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmax_pool_3d_grad_with_argmax.cc232 auto output_shape_vec = input_grads_info->GetTensorShape()->GetDimSizes(); in MaxPool3DGradWithArgmaxCompute() local
Dmax_pool_3d_with_argmax.cc259 auto output_shape_vec = output_y_info->GetTensorShape()->GetDimSizes(); in MaxPool3DWithArgmaxCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/customize/op_proto/
Dnn_pooling_ops_proto.cc319 std::vector<int64_t> output_shape_vec(kRank, ge::UNKNOWN_DIM); in CUST_IMPLEMT_INFERFUNC() local
393 …std::vector<int64_t> output_shape_vec{input_shape[0], input_shape[1], output_d, output_h, output_w… in CUST_IMPLEMT_INFERFUNC() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/kernels/
Daudio_utils.cc351 auto output_shape_vec = input_shape.AsVector(); in TimeStretch() local
2473 std::vector<int64_t> output_shape_vec = st_shape.AsVector(); in LFCC() local
2545 std::vector<int64_t> output_shape_vec = st_shape.AsVector(); in MFCC() local