Struct QuantizedConv2D

pub struct QuantizedConv2D { /* private fields */ }

Builder for the QuantizedConv2D operation.

Implementations

impl QuantizedConv2D

pub fn new() -> Self

Creates a new QuantizedConv2D.

pub fn Tinput<ArgType: Into<DataType>>(self, value: ArgType) -> Self

Sets the Tinput attribute.

pub fn Tfilter<ArgType: Into<DataType>>(self, value: ArgType) -> Self

Sets the Tfilter attribute.

pub fn out_type<ArgType: Into<DataType>>(self, value: ArgType) -> Self

Sets the out_type attribute.

pub fn strides<ArgType: Into<Vec<i64>>>(self, value: ArgType) -> Self

Sets the strides attribute.

pub fn padding<ArgType: Into<String>>(self, value: ArgType) -> Self

Sets the padding attribute.

pub fn dilations<ArgType: Into<Vec<i64>>>(self, value: ArgType) -> Self

Sets the dilations attribute.

pub fn add_control_input(self, op: Operation) -> Self

Adds a control input.

pub fn build<O0: Into<Output>, O1: Into<Output>, O2: Into<Output>, O3: Into<Output>, O4: Into<Output>, O5: Into<Output>>( &self, input: O0, filter: O1, min_input: O2, max_input: O3, min_filter: O4, max_filter: O5, scope: &mut Scope, ) -> Result<Operation>

Builds the QuantizedConv2D operation.

Trait Implementations

impl Debug for QuantizedConv2D

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for QuantizedConv2D

fn default() -> QuantizedConv2D

Returns the “default value” for a type.