apollo::perception::inference::dfmb_psroi_align_op结构体 参考

#include <perception_inference_migraphx_plugins.h>

apollo::perception::inference::dfmb_psroi_align_op 的协作图:

Public 成员函数
std::string	name () const
migraphx::shape	compute_shape (const std::vector< migraphx::shape > &inputs) const
migraphx::argument	compute (migraphx::context &gctx, const migraphx::shape &, const std::vector< migraphx::argument > &args) const
int	output_alias (const std::vector< migraphx::shape > &shapes) const

静态 Public 成员函数
template<class Self , class F >
static auto	reflect (Self &self, F f)

Public 属性
size_t	batch_size
std::function< std::shared_ptr< DFMBPSROIAlignPlugin >()>	get_plugin {}

详细描述

在文件 perception_inference_migraphx_plugins.h 第 34 行定义.

成员函数说明

compute()

migraphx::argument apollo::perception::inference::dfmb_psroi_align_op::compute ( migraphx::context &  gctx, const migraphx::shape &  , const std::vector< migraphx::argument > &  args  ) const

inline

在文件 perception_inference_migraphx_plugins.h 第 59 行定义.

                                                     {
    CHECK_GE(static_cast<int>(args.size()), 3);
    bool no_trans = args.size() < 4;  // 2 imputs + output
 
    auto bottom_data = args[0];
    auto bottom_rois = args[1];
    auto bottom_trans = no_trans ? migraphx::argument{} : args[2];
    auto top_data = no_trans ? args[2] : args[3];
 
    std::vector<const void*> inputs_data{
        bottom_data.data(), bottom_rois.data(),
        no_trans ? nullptr : bottom_trans.data()};
 
    std::vector<void*> outputs_data{top_data.data()};
 
    auto& ctx = migraphx::any_cast<migraphx::gpu::context>(gctx);
 
    auto plugin = get_plugin();
    plugin->enqueue(batch_size, inputs_data.data(), outputs_data.data(),
                    nullptr /*workspace*/, ctx.get_stream().get());
    return top_data;
  }

compute_shape()

migraphx::shape apollo::perception::inference::dfmb_psroi_align_op::compute_shape ( const std::vector< migraphx::shape > &  inputs ) const

inline

在文件 perception_inference_migraphx_plugins.h 第 44 行定义.

                                                    {
    migraphx::check_shapes{inputs, *this}.standard();
 
    CHECK_EQ(static_cast<int>(batch_size), 1);
 
    auto plugin = get_plugin();
    auto out_dims = plugin->getOutputDimensions(0, nullptr, 0);
 
    return migraphx::shape{
        migraphx::shape::float_type,
        {std::size_t(out_dims.d[0]), std::size_t(out_dims.d[1]),
         std::size_t(out_dims.d[2]), std::size_t(out_dims.d[3])}};
  }

name()

std::string apollo::perception::inference::dfmb_psroi_align_op::name ( ) const

inline

在文件 perception_inference_migraphx_plugins.h 第 42 行定义.

{ return "dfmb_psroi_align"; }

output_alias()

int apollo::perception::inference::dfmb_psroi_align_op::output_alias ( const std::vector< migraphx::shape > &  shapes ) const

inline

在文件 perception_inference_migraphx_plugins.h 第 85 行定义.

                                                                 {
    return shapes.size() - 1;
  }

reflect()

template<class Self , class F >

static auto apollo::perception::inference::dfmb_psroi_align_op::reflect ( Self &  self, F  f  )

inlinestatic

在文件 perception_inference_migraphx_plugins.h 第 38 行定义.

                                       {  // NOLINT
    return migraphx::pack(f(self.batch_size, "batch_size"));
  }

类成员变量说明

batch_size

size_t apollo::perception::inference::dfmb_psroi_align_op::batch_size

在文件 perception_inference_migraphx_plugins.h 第 35 行定义.

get_plugin

std::function<std::shared_ptr<DFMBPSROIAlignPlugin>()> apollo::perception::inference::dfmb_psroi_align_op::get_plugin {}

在文件 perception_inference_migraphx_plugins.h 第 36 行定义.

{};

---

该结构体的文档由以下文件生成:

• modules/perception/common/inference/migraphx/plugins/perception_inference_migraphx_plugins.h