Registers a dataset with the tf.data service.
tf.raw_ops.RegisterDataset(
dataset, address, protocol, external_state_policy, name=None
)
Args | |
---|---|
dataset
|
A Tensor of type variant .
|
address
|
A Tensor of type string .
|
protocol
|
A Tensor of type string .
|
external_state_policy
|
An int .
|
name
|
A name for the operation (optional). |
Returns | |
---|---|
A Tensor of type int64 .
|