Home
last modified time | relevance | path

Searched defs:output_tensor (Results 1 – 25 of 94) sorted by relevance

1234

/third_party/mindspore/tests/ut/cpp/dataset/
Drandom_color_adjust_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
47 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
59 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
71 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dcenter_crop_op_test.cc34 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
48 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
61 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dnormalizepad_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
50 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Drandom_color_op_test.cc76 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
84 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
93 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Drandom_crop_and_resize_op_test.cc38 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
64 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
90 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dequalize_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dinvert_op_test.cc33 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dauto_contrast_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Ddecode_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Drandom_affine_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Drescale_op_test.cc37 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dresize_op_test.cc37 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Drandom_sharpness_op_test.cc42 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dchannel_swap_test.cc39 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
/third_party/mindspore/mindspore/ccsrc/minddata/dataset/kernels/image/
Dlite_image_utils.cc211 std::shared_ptr<Tensor> output_tensor; in JpegCropAndDecode() local
290 std::shared_ptr<Tensor> output_tensor; in Crop() local
357 std::shared_ptr<Tensor> output_tensor; in Normalize() local
401 std::shared_ptr<Tensor> output_tensor; in Resize() local
475 std::shared_ptr<Tensor> output_tensor; in RgbToBgr() local
509 std::shared_ptr<Tensor> output_tensor; in RgbToGray() local
550 std::shared_ptr<Tensor> output_tensor; in Pad() local
626 std::shared_ptr<Tensor> output_tensor; in RotateAngleWithOutMirror() local
696 std::shared_ptr<Tensor> output_tensor; in RotateAngleWithMirror() local
761 std::shared_ptr<Tensor> output_tensor; in Affine() local
[all …]
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32/
Dspace_to_batch_fp32.cc30 auto output_tensor = out_tensors_.at(FIRST_INPUT); in ProcessInput() local
74 auto output_tensor = out_tensors_.at(FIRST_INPUT); in ReSize() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dconcat_int8.cc56 auto output_tensor = out_tensors_.at(kOutputIndex); in Init() local
101 auto output_tensor = out_tensors_.at(kOutputIndex); in ReSize() local
Dspace_to_batch_int8.cc30 auto output_tensor = out_tensors_.at(0); in Run() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/sponge/pme/
Difft_3d_impl.cu20 void IFFT3D(int Nfft, Complex<T> *input_tensor, T *output_tensor, const cufftHandle &FFT_plan_c2r, in IFFT3D()
Dfft_3d_impl.cu20 void FFT3D(int Nfft, T *input_tensor, Complex<T> *output_tensor, const cufftHandle &FFT_plan_r2c, c… in FFT3D()
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dstrided_slice_fp32_tests.cc146 lite::Tensor output_tensor; in TEST_F() local
200 lite::Tensor output_tensor; in TEST_F() local
261 lite::Tensor output_tensor; in TEST_F() local
322 lite::Tensor output_tensor; in TEST_F() local
375 lite::Tensor output_tensor; in TEST_F() local
436 lite::Tensor output_tensor; in TEST_F() local
591 lite::Tensor output_tensor; in TEST_F() local
/third_party/mindspore/mindspore/lite/tools/converter/quantizer/quant_helper/
Dquant_dtype_cast_quant_param_propogator.cc31 auto &output_tensor = graph->allTensors.at(node.outputIndex.at(0)); in PropogateQuantParams() local
Dmatmul_quant_type_determiner.cc30 auto &output_tensor = graph.allTensors.at(node->outputIndex.at(kOutputIndex)); in DetermineQuantWeight() local
Dconv_quant_type_determiner.cc30 auto &output_tensor = graph.allTensors.at(node->outputIndex.at(kOutputIndex)); in DetermineQuantWeight() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dquant_dtype_cast_tests.cc57 lite::Tensor output_tensor; in TEST_F() local
107 lite::Tensor output_tensor; in TEST_F() local

1234