Home
last modified time | relevance | path

Searched defs:align_corners (Results 1 – 25 of 30) sorted by relevance

12

/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dresize_nearest_neighbor_fp32_tests.cc51 … float *input_data, float *output_data, const bool align_corners, in Prepare()
81 bool align_corners = false; in TEST_F() local
98 bool align_corners = false; in TEST_F() local
115 bool align_corners = false; in TEST_F() local
132 bool align_corners = false; in TEST_F() local
149 bool align_corners = false; in TEST_F() local
166 bool align_corners = false; in TEST_F() local
183 bool align_corners = false; in TEST_F() local
200 bool align_corners = false; in TEST_F() local
217 bool align_corners = false; in TEST_F() local
[all …]
Dresize_bilinear_fp32_tests.cc54 … float *input_data, float *output_data, const bool align_corners, in Prepare()
85 bool align_corners = false; in TEST_F() local
102 bool align_corners = false; in TEST_F() local
119 bool align_corners = false; in TEST_F() local
136 bool align_corners = false; in TEST_F() local
153 bool align_corners = false; in TEST_F() local
170 bool align_corners = false; in TEST_F() local
187 bool align_corners = false; in TEST_F() local
204 bool align_corners = false; in TEST_F() local
221 bool align_corners = false; in TEST_F() local
[all …]
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/opencl/
Dresize_tests.cc25 … *CreateParameter(schema::ResizeMethod method, int new_height, int new_width, bool align_corners) { in CreateParameter()
42 bool align_corners = false; in TEST_F() local
58 bool align_corners = true; in TEST_F() local
74 bool align_corners = false; in TEST_F() local
90 bool align_corners = false; in TEST_F() local
107 bool align_corners = true; in TEST_F() local
123 bool align_corners = false; in TEST_F() local
/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dresize_bilinear_int8_tests.cc57 … const mindspore::LiteQuantParam quant_out, const bool align_corners, in Prepare()
95 bool align_corners = false; in TEST_F() local
115 bool align_corners = false; in TEST_F() local
141 bool align_corners = true; in TEST_F() local
167 bool align_corners = true; in TEST_F() local
Dresize_nearest_neighbor_int8_tests.cc52 … const LiteQuantParam quant_out, const bool align_corners, in Prepare()
/third_party/mindspore/mindspore/core/ops/
Dresize_nearest_neighbor.cc29 void ResizeNearestNeighbor::Init(const std::vector<int64_t> &size, const bool align_corners) { in Init()
34 void ResizeNearestNeighbor::set_align_corners(const bool align_corners) { in set_align_corners()
Dresize_bilinear.cc32 void ResizeBilinear::set_align_corners(const bool align_corners) { in set_align_corners()
41 void ResizeBilinear::Init(const std::vector<int64_t> &size, const bool align_corners) { in Init()
/third_party/mindspore/mindspore/core/ops/grad/
Dresize_grad.cc28 void ResizeGrad::Init(const ResizeMethod method, const bool align_corners) { in Init()
38 void ResizeGrad::set_align_corners(const bool align_corners) { in set_align_corners()
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/
Dresize_grad.cc32 float Scaling(size_t in_size, size_t out_size, bool align_corners) { in Scaling()
43 bool align_corners = param->align_corners_; in ReSize() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp16_grad/
Dresize_fp16_grad.cc32 float16_t ScalingFp16(size_t in_size, size_t out_size, bool align_corners) { in ScalingFp16()
43 bool align_corners = param->align_corners_; in ReSize() local
/third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/int8/
Dresize_int8.cc157 bool align_corners = coordinate_transform_mode_ == schema::CoordinateTransformMode_ALIGN_CORNERS; in CalRatio() local
229 bool align_corners = coordinate_transform_mode_ == schema::CoordinateTransformMode_ALIGN_CORNERS; in CalFloatRatio() local
358 … bool align_corners = coordinate_transform_mode_ == schema::CoordinateTransformMode_ALIGN_CORNERS; in RunImpl() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/
Dresize_nearest_neighbor_impl.cu26 bool align_corners, float h_scale, float w_scale) { in ResizeNearestNeighbor()
61 … T *output, const int d1, const int d2, const int d3, const int d4, bool align_corners, in CalResizeNearestNeighbor()
Dresize_nearest_neighbor_grad_impl.cu34 … const int d3, const int d4, bool align_corners, float h_scale, in ResizeNearestNeighborGrad()
71bool align_corners, float h_scale, float w_scale, cudaStream_t cuda_stream) { in CalResizeNearestNeighborGrad()
/third_party/mindspore/tests/st/ops/gpu/
Dtest_resize_bilinear_grad_op.py26 def __init__(self, align_corners=False): argument
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/
Dresize_int8_coder.cc90 … bool align_corners = coordinate_transform_mode_ == schema::CoordinateTransformMode_ALIGN_CORNERS; in DoCode() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/
Dresize_int8.c160 … const int *output_shape, const bool align_corners, int tid, int thread_num) { in ResizeNearestNeighborInt8Simple()
186 …NeighborInt(const int32_t pos, const int in_size, const int32_t new_size, const bool align_corners, in ComputeNearestNeighborInt()
199 … const int *output_shape, const bool align_corners, const QuantMulArg *multiplier, in ResizeNearestNeighborInt8()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/arrays/
Dresize_nearest_neighbor_grad_gpu_kernel.h120 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
Dresize_nearest_neighbor_gpu_kernel.h112 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dresize_bilinear_gpu_kernel.h118 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
Dresize_bilinear_grad_gpu_kernel.h136 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
/third_party/mindspore/tests/st/ops/cpu/
Dtest_resize_nearest_neighbor_op.py25 def __init__(self, size=None, align_corners=False): argument
/third_party/mindspore/mindspore/nn/layer/
Dbasic.py829 def bilinear(shape, size, scale, align_corners, prim_name=None): argument
908 def construct(self, x, size=None, scale_factor=None, align_corners=False): argument
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/
Dresize_grad.c23 bool align_corners = param->align_corners_; in ResizeNearestNeighborGrad() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/
Dresize_grad.c22 bool align_corners = param->align_corners_; in ResizeNearestNeighborFp16Grad() local
/third_party/mindspore/mindspore/lite/tools/converter/import/
Dprimitive_adjust.cc536 auto align_corners = GetValue<bool>(src_prim->GetAttr(ops::kAlignCorners)); in MoveAttrMapResizeGrad() local

12