Home
last modified time | relevance | path

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

1234567

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/dataset/
Dto_tensor_op_test.cc43 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
54 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
65 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
76 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
87 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
98 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
109 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
120 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
131 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
142 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
[all …]
Drandom_color_adjust_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
52 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
69 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
86 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
51 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
67 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
65 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
92 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dnormalizepad_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
53 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
87 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
99 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Dnormalize_op_test.cc36 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
65 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Ddecode_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
Dequalize_op_test.cc35 std::shared_ptr<Tensor> output_tensor; in TEST_F() local
Deager_auto_contrast_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
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/customize/
Dreshape.cc37 auto output_tensor = view_op->Call(input_tensor, shape); in ReshapeCustomize() local
47 auto output_tensor = view_op->Call(copy_tensor, shape); in ReshapeCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/int8/
Dconcat_int8.cc68 auto output_tensor = out_tensors_.at(kOutputIndex); in Prepare() local
100 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-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/kernel/
Dscale.c122 TensorC *output_tensor = scale->base_.out_[OUTPUT_INDEX]; in ScaleCalculateParameter() local
209 TensorC *output_tensor = self->out_[OUTPUT_INDEX]; in ScaleCheckInputsOutputs() local
272 TensorC *output_tensor = self->out_[OUTPUT_INDEX]; in ScaleCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/
Dlite_image_utils.cc276 std::shared_ptr<Tensor> output_tensor; in JpegCropAndDecode() local
386 std::shared_ptr<Tensor> output_tensor; in Crop() local
457 std::shared_ptr<Tensor> output_tensor; in Normalize() local
521 std::shared_ptr<Tensor> output_tensor; in Resize() local
593 std::shared_ptr<Tensor> output_tensor; in Resize() local
674 std::shared_ptr<Tensor> output_tensor; in RgbToBgr() local
711 std::shared_ptr<Tensor> output_tensor; in RgbToGray() local
757 std::shared_ptr<Tensor> output_tensor; in Pad() local
839 std::shared_ptr<Tensor> output_tensor; in RotateAngleWithOutMirror() local
914 std::shared_ptr<Tensor> output_tensor; in RotateAngleWithMirror() local
[all …]
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/
Dcontiguous.cc26 auto output_tensor = ContiguousTensorOpProcess(op, input_tensor); in ContiguousGPUCustomize() local
Dclamp_tensor.cc29 auto output_tensor = ClampTensorCustomizeCall(op, x_tensor, min, max, "GPU"); in ClampTensorGPUCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/
Dcontiguous.cc26 auto output_tensor = ContiguousTensorOpProcess(op, input_tensor); in ContiguousCPUCustomize() local
Dclamp_tensor.cc29 auto output_tensor = ClampTensorCustomizeCall(op, x_tensor, min, max, "CPU"); in ClampTensorCPUCustomize() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/aicpu_ops/cpu_kernel/ms_kernel/
Dmatmul.cc50 auto output_tensor = ctx.Output(kFirstOutputIndex); in BiasCompute() local
104 auto output_tensor = ctx.Output(kFirstOutputIndex); in MatMulCompute() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dcontiguous.cc31 auto output_tensor = ContiguousTensorOpProcess(op, input_tensor); in ContiguousAscendCustomize() local
/third_party/mindspore/mindspore-src/source/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
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/kernel_builder/ascend/tbe_and_aicpu/cpukernel/impl/
Dreshape_cust_kernels.cc21 Tensor *output_tensor = ctx.Output(0); in Compute() local

1234567