Home
last modified time | relevance | path

Searched defs:interpolation_mode (Results 1 – 21 of 21) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/
Dgrid_sampler_2d_grad_aclnn_kernel.cc31 auto interpolation_mode = transform::ConvertKernelTensor<int64_t>(inputs[kIndex3]); in GetWorkSpaceInfo() local
42 auto interpolation_mode = transform::ConvertKernelTensor<int64_t>(inputs[kIndex3]); in Launch() local
Dgrid_sampler_3d_grad_aclnn_kernel.cc31 auto interpolation_mode = transform::ConvertKernelTensor<int64_t>(inputs[kIndex3]); in GetWorkSpaceInfo() local
42 auto interpolation_mode = transform::ConvertKernelTensor<int64_t>(inputs[kIndex3]); in Launch() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_grid_sampler_3d.cc35 ValuePtr interpolation_mode; member
59 auto interpolation_mode = shape_param.interpolation_mode->ToAbstract(); in TEST_P() local
Dtest_ops_grid_sampler_2d.cc35 ValuePtr interpolation_mode; member
59 auto interpolation_mode = shape_param.interpolation_mode->ToAbstract(); in TEST_P() local
Dtest_ops_grid_sampler_2d_grad.cc36 ValuePtr interpolation_mode; member
65 auto interpolation_mode = shape_param.interpolation_mode->ToAbstract(); in TEST_P() local
Dtest_ops_grid_sampler_3d_grad.cc36 ValuePtr interpolation_mode; member
65 auto interpolation_mode = shape_param.interpolation_mode->ToAbstract(); in TEST_P() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dgrid_sampler_3d_grad_cpu_kernel.h73 int64_t interpolation_mode; variable
Dgrid_sampler_2d_cpu_kernel.cc200 … const int64_t *grid_stride, int64_t interpolation_mode, int64_t padding_mode, in Call2Half()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/
Dgrid_sampler_2d_grad.cc32 …const BaseTensorPtr &grid_tensor, const Int64ImmPtr &interpolation_mode, const Int64ImmPtr &paddin… in GridSampler2DGradAscendCustomize()
Dgrid_sampler_3d_grad.cc32 …const BaseTensorPtr &grid_tensor, const Int64ImmPtr &interpolation_mode, const Int64ImmPtr &paddin… in GridSampler3DGradAscendCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/cuda_impl/cuda_ops/
Dgrid_sampler_impl.cu33 GridSamplerInterpolationMode interpolation_mode, in GridSampler2DKernel()
140 const GridSamplerInterpolationMode interpolation_mode, in GridSampler2D()
183 GridSamplerInterpolationMode interpolation_mode, in GridSampler3DKernel()
309 const GridSamplerInterpolationMode interpolation_mode, in GridSampler3D()
Dgrid_sampler_grad_impl.cu71 … const size_t dgrid_sW, GridSamplerInterpolationMode interpolation_mode, in GridSampler2DGradKernel()
234 const GridSamplerInterpolationMode interpolation_mode, in GridSampler2DGrad()
290 const size_t grad_sW, const size_t dgrid_sW, GridSamplerInterpolationMode interpolation_mode, in GridSampler3DGradKernel()
478 const GridSamplerInterpolationMode interpolation_mode, in GridSampler3DGrad()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/ascend910b/
Ddvpp_affine_op.cc69 uint32_t interpolation_mode = GetDVPPInterpolationMode(interpolation_); in Compute() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/mapper/
Dresize_mapper.cc189 auto interpolation_mode = static_cast<ResizeMethod>(resize_prim->get_method()); in Map() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/checker/
Dresize_checker.cc97 …auto interpolation_mode = static_cast<ResizeMethod>(api::GetValue<int64_t>(prim->GetAttr(ops::kMet… in CheckResizeOp() local
/third_party/mesa3d/src/microsoft/compiler/
Ddxil_signature.h78 uint8_t interpolation_mode; // DXIL::InterpolationMode or D3D10_SB_INTERPOLATION_MODE member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/api/python/bindings/dataset/kernels/ir/image/
Dbindings.cc706 const std::vector<float> &center, const std::vector<uint8_t> &fill_value) { in __anonfa8cf3896b02()
860 … .def(py::init([](const std::vector<int32_t> &size, InterpolationMode interpolation_mode) { in __anonfa8cf3898302()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc905 void GridSampler2DGrad::set_interpolation_mode(const int64_t &interpolation_mode) { (vo… in set_interpolation_mode()
920 void GridSampler2D::set_interpolation_mode(const int64_t &interpolation_mode) { (void)t… in set_interpolation_mode()
935 void GridSampler3DGrad::set_interpolation_mode(const int64_t &interpolation_mode) { (vo… in set_interpolation_mode()
950 void GridSampler3D::set_interpolation_mode(const int64_t &interpolation_mode) { (void)t… in set_interpolation_mode()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/utils/
Dacl_plugin.cc346 … const std::vector<float> &matrix, uint32_t interpolation_mode, uint32_t padding_mode, in PluginDvppAffine()
Ddvpp_image_utils.cc529 … uint32_t interpolation_mode, uint32_t padding_mode, const std::vector<float> &fill) { in DvppAffine()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/kernels/image/dvpp/
Dacl_adapter.cc553 … uint32_t interpolation_mode, uint32_t padding_mode, const std::vector<float> &fill) { in DvppAffine()