Struct tensorflow::ops::TPUCompile

pub struct TPUCompile { /* private fields */ }

Builder for the TPUCompile operation.

Implementations

impl TPUCompile

pub fn new() -> Self

Creates a new TPUCompile.

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

Sets the num_computations attribute.

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

Sets the function attribute.

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

Sets the metadata attribute.

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

Sets the NumDynamicShapes attribute.

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

Sets the Tguaranteed_constants attribute.

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

Adds a control input.

pub fn build<O0: Into<Output>, O1: Into<Output>>( &self, dynamic_shapes: O0, guaranteed_constants: O1, scope: &mut Scope ) -> Result<Operation>

Builds the TPUCompile operation.

pub fn build_instance( &self, dynamic_shapes: Vec<Output>, guaranteed_constants: Output, scope: &mut Scope ) -> Result<TPUCompileInst>

Builds a new instance of ‘TPUCompile’ Operation with it’s Outputs and Inputs exposed as methods.

Trait Implementations

impl Debug for TPUCompile

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

Formats the value using the given formatter.

impl Default for TPUCompile

fn default() -> TPUCompile

Returns the “default value” for a type.