Home
last modified time | relevance | path

Searched defs:tensor_op (Results 1 – 25 of 111) sorted by relevance

12345

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/text/ir/kernels/
Dtext_ir.cc70 std::shared_ptr<AddTokenOp> tensor_op = std::make_shared<AddTokenOp>(token_, begin_); in Build() local
114 std::shared_ptr<BasicTokenizerOp> tensor_op = std::make_shared<BasicTokenizerOp>( in Build() local
160 std::shared_ptr<BertTokenizerOp> tensor_op = in Build() local
170 std::shared_ptr<CaseFoldOp> tensor_op = std::make_shared<CaseFoldOp>(); in Build() local
180 std::shared_ptr<FilterWikipediaXMLOp> tensor_op = std::make_shared<FilterWikipediaXMLOp>(); in Build() local
320 std::shared_ptr<JiebaTokenizerOp> tensor_op = in Build() local
377 …std::shared_ptr<LookupOp> tensor_op = std::make_shared<LookupOp>(vocab_, default_id_, DataType(dat… in Build() local
421 …std::shared_ptr<NgramOp> tensor_op = std::make_shared<NgramOp>(ngrams_, l_len, l_pad, r_len, r_pad… in Build() local
440 std::shared_ptr<NormalizeUTF8Op> tensor_op = std::make_shared<NormalizeUTF8Op>(normalize_form_); in Build() local
451 …std::shared_ptr<RegexReplaceOp> tensor_op = std::make_shared<RegexReplaceOp>(pattern_, replace_, r… in Build() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/ir/vision/
Dascend_vision_ir.cc71 …std::shared_ptr<DvppCropJpegOp> tensor_op = std::make_shared<DvppCropJpegOp>(cropHeight, cropWidth… in Build() local
127 std::shared_ptr<DvppDecodeResizeJpegOp> tensor_op = in Build() local
208 auto tensor_op = std::make_shared<DvppDecodeVideoOp>(width, height, en_type_, format_, output_); in Build() local
326 std::shared_ptr<DvppDecodeResizeCropJpegOp> tensor_op = in Build() local
404 std::shared_ptr<DvppNormalizeOp> tensor_op = std::make_shared<DvppNormalizeOp>(mean_, std_); in Build() local
471 …std::shared_ptr<DvppResizeJpegOp> tensor_op = std::make_shared<DvppResizeJpegOp>(resizeHeight, res… in Build() local
Drandom_select_subpolicy_ir.cc79 …std::shared_ptr<RandomSelectSubpolicyOp> tensor_op = std::make_shared<RandomSelectSubpolicyOp>(pol… in Build() local
90 auto tensor_op = op_pair.first; in to_json() local
Drgba_to_rgb_ir.cc36 std::shared_ptr<RgbaToRgbOp> tensor_op = std::make_shared<RgbaToRgbOp>(); in Build() local
Dswap_red_blue_ir.cc36 std::shared_ptr<SwapRedBlueOp> tensor_op = std::make_shared<SwapRedBlueOp>(); in Build() local
Drgba_to_bgr_ir.cc36 std::shared_ptr<RgbaToBgrOp> tensor_op = std::make_shared<RgbaToBgrOp>(); in Build() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/audio/ir/kernels/
Dangle_ir.cc30 std::shared_ptr<AngleOp> tensor_op = std::make_shared<AngleOp>(); in Build() local
Dcontrast_ir.cc34 std::shared_ptr<ContrastOp> tensor_op = std::make_shared<ContrastOp>(enhancement_amount_); in Build() local
Driaa_biquad_ir.cc33 std::shared_ptr<RiaaBiquadOp> tensor_op = std::make_shared<RiaaBiquadOp>(sample_rate_); in Build() local
Dgain_ir.cc30 std::shared_ptr<GainOp> tensor_op = std::make_shared<GainOp>(gain_db_); in Build() local
Ddc_shift_ir.cc34 std::shared_ptr<DCShiftOp> tensor_op = std::make_shared<DCShiftOp>(shift_, limiter_gain_); in Build() local
Doverdrive_ir.cc34 std::shared_ptr<OverdriveOp> tensor_op = std::make_shared<OverdriveOp>(gain_, color_); in Build() local
Ddb_to_amplitude_ir.cc30 std::shared_ptr<DBToAmplitudeOp> tensor_op = std::make_shared<DBToAmplitudeOp>(ref_, power_); in Build() local
Dmu_law_encoding_ir.cc43 …std::shared_ptr<MuLawEncodingOp> tensor_op = std::make_shared<MuLawEncodingOp>(quantization_channe… in Build() local
Ddeemph_biquad_ir.cc38 std::shared_ptr<DeemphBiquadOp> tensor_op = std::make_shared<DeemphBiquadOp>(sample_rate_); in Build() local
Dmu_law_decoding_ir.cc44 …std::shared_ptr<MuLawDecodingOp> tensor_op = std::make_shared<MuLawDecodingOp>(quantization_channe… in Build() local
Dcomplex_norm_ir.cc43 std::shared_ptr<ComplexNormOp> tensor_op = std::make_shared<ComplexNormOp>(power_); in Build() local
Ddither_ir.cc33 …std::shared_ptr<DitherOp> tensor_op = std::make_shared<DitherOp>(density_function_, noise_shaping_… in Build() local
Dallpass_biquad_ir.cc37 …std::shared_ptr<AllpassBiquadOp> tensor_op = std::make_shared<AllpassBiquadOp>(sample_rate_, centr… in Build() local
Dvol_ir.cc43 std::shared_ptr<VolOp> tensor_op = std::make_shared<VolOp>(gain_, gain_type_); in Build() local
Dbandreject_biquad_ir.cc36 …std::shared_ptr<BandrejectBiquadOp> tensor_op = std::make_shared<BandrejectBiquadOp>(sample_rate_,… in Build() local
Dlfilter_ir.cc38 std::shared_ptr<LFilterOp> tensor_op = std::make_shared<LFilterOp>(a_coeffs_, b_coeffs_, clamp_); in Build() local
Dhighpass_biquad_ir.cc36 …std::shared_ptr<HighpassBiquadOp> tensor_op = std::make_shared<HighpassBiquadOp>(sample_rate_, cut… in Build() local
Dlowpass_biquad_ir.cc36 …std::shared_ptr<LowpassBiquadOp> tensor_op = std::make_shared<LowpassBiquadOp>(sample_rate_, cutof… in Build() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Duniform_aug_op.cc38 for (const auto &tensor_op : selected_tensor_ops) { in Compute() local

12345