ONE - On-device Neural Engine
Loading...
Searching...
No Matches
tfldump::StridedSlicePrinter Class Reference
Collaboration diagram for tfldump::StridedSlicePrinter:

Public Member Functions

void options (const tflite::Operator *op, std::ostream &os) const override
 

Detailed Description

Definition at line 557 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

void tfldump::StridedSlicePrinter::options ( const tflite::Operator *  op,
std::ostream &  os 
) const
inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 560 of file OpPrinter.cpp.

561 {
562 if (auto *strided_slice_params = op->builtin_options_as_StridedSliceOptions())
563 {
564 os << " ";
565 os << "begin_mask(" << strided_slice_params->begin_mask() << ") ";
566 os << "end_mask(" << strided_slice_params->end_mask() << ") ";
567 os << "ellipsis_mask(" << strided_slice_params->ellipsis_mask() << ") ";
568 os << "new_axis_mask(" << strided_slice_params->new_axis_mask() << ") ";
569 os << "shrink_axis_mask(" << strided_slice_params->shrink_axis_mask() << ") ";
570 os << std::endl;
571 }
572 }

The documentation for this class was generated from the following file: