Public Member Functions | |
def | __init__ |
def | create_operator |
def | declare_backward_dependency |
def | infer_shape |
def | list_arguments |
def | list_outputs |
Private Attributes | |
_num_classes | |
_num_reg_classes | |
_roi_per_img |
Definition at line 62 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMProp.__init__ | ( | self, | |
num_classes, | |||
num_reg_classes, | |||
roi_per_img | |||
) |
Definition at line 63 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMProp.create_operator | ( | self, | |
ctx, | |||
shapes, | |||
dtypes | |||
) |
Definition at line 82 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMProp.declare_backward_dependency | ( | self, | |
out_grad, | |||
in_data, | |||
out_data | |||
) |
Definition at line 85 of file box_annotator_ohem.py.
def operator_py.box_annotator_ohem.BoxAnnotatorOHEMProp.infer_shape | ( | self, | |
in_shape | |||
) |
Definition at line 75 of file box_annotator_ohem.py.
Definition at line 69 of file box_annotator_ohem.py.
Definition at line 72 of file box_annotator_ohem.py.
Definition at line 63 of file box_annotator_ohem.py.
Definition at line 63 of file box_annotator_ohem.py.
Definition at line 63 of file box_annotator_ohem.py.