/third_party/mindspore/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/ |
D | resize_nearest_neighbor_fp32_tests.cc | 51 … 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 …]
|
D | resize_bilinear_fp32_tests.cc | 54 … 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/ |
D | resize_tests.cc | 25 … *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/ |
D | resize_bilinear_int8_tests.cc | 57 … 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
|
D | resize_nearest_neighbor_int8_tests.cc | 52 … const LiteQuantParam quant_out, const bool align_corners, in Prepare()
|
/third_party/mindspore/mindspore/core/ops/ |
D | resize_nearest_neighbor.cc | 29 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()
|
D | resize_bilinear.cc | 32 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/ |
D | resize_grad.cc | 28 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/ |
D | resize_grad.cc | 32 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/ |
D | resize_fp16_grad.cc | 32 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/ |
D | resize_int8.cc | 157 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/ |
D | resize_nearest_neighbor_impl.cu | 26 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()
|
D | resize_nearest_neighbor_grad_impl.cu | 34 … const int d3, const int d4, bool align_corners, float h_scale, in ResizeNearestNeighborGrad() 71 … bool align_corners, float h_scale, float w_scale, cudaStream_t cuda_stream) { in CalResizeNearestNeighborGrad()
|
/third_party/mindspore/tests/st/ops/gpu/ |
D | test_resize_bilinear_grad_op.py | 26 def __init__(self, align_corners=False): argument
|
/third_party/mindspore/mindspore/lite/micro/coder/opcoders/nnacl/int8/ |
D | resize_int8_coder.cc | 90 … bool align_corners = coordinate_transform_mode_ == schema::CoordinateTransformMode_ALIGN_CORNERS; in DoCode() local
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/int8/ |
D | resize_int8.c | 160 … 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/ |
D | resize_nearest_neighbor_grad_gpu_kernel.h | 120 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
|
D | resize_nearest_neighbor_gpu_kernel.h | 112 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/ |
D | resize_bilinear_gpu_kernel.h | 118 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
|
D | resize_bilinear_grad_gpu_kernel.h | 136 float Scaling(const int in_size, const int out_size, bool align_corners) { in Scaling()
|
/third_party/mindspore/tests/st/ops/cpu/ |
D | test_resize_nearest_neighbor_op.py | 25 def __init__(self, size=None, align_corners=False): argument
|
/third_party/mindspore/mindspore/nn/layer/ |
D | basic.py | 829 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/ |
D | resize_grad.c | 23 bool align_corners = param->align_corners_; in ResizeNearestNeighborGrad() local
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
D | resize_grad.c | 22 bool align_corners = param->align_corners_; in ResizeNearestNeighborFp16Grad() local
|
/third_party/mindspore/mindspore/lite/tools/converter/import/ |
D | primitive_adjust.cc | 536 auto align_corners = GetValue<bool>(src_prim->GetAttr(ops::kAlignCorners)); in MoveAttrMapResizeGrad() local
|