Struct tensorflow::ops::XlaHostCompute

pub struct XlaHostCompute { /* private fields */ }

Builder for the XlaHostCompute operation.

Implementations

impl XlaHostCompute

pub fn new() -> Self

Creates a new XlaHostCompute.

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

Sets the Tinputs attribute.

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

Sets the Toutputs attribute.

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

Sets the ancestors attribute.

pub fn shapes<ArgType: Into<Vec<Shape>>>(self, value: ArgType) -> Self

Sets the shapes attribute.

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

Sets the shape_inference_graph attribute.

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

Sets the key attribute.

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

Sets the send_key attribute.

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

Sets the recv_key attribute.

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

Sets the cost_estimate_ns attribute.

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

Sets the tpu_core attribute.

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

Adds a control input.

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

Builds the XlaHostCompute operation.

pub fn build_instance( &self, inputs: Output, scope: &mut Scope ) -> Result<XlaHostComputeInst>

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

Trait Implementations

impl Debug for XlaHostCompute

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

Formats the value using the given formatter.

impl Default for XlaHostCompute

fn default() -> XlaHostCompute

Returns the “default value” for a type.