Home
last modified time | relevance | path

Searched refs:output0_type (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dcustom_extract_features.cc29 auto output0_type = kInt32; in CustomExtractFeaturesInfer() local
42 auto output0 = std::make_shared<abstract::AbstractTensor>(output0_type, out_shape); in CustomExtractFeaturesInfer()
Dtopk.cc44 auto output0_type = in TopKInfer() local
53 auto output0 = std::make_shared<abstract::AbstractTensor>(output0_type, x_shape); in TopKInfer()