View source on GitHub |
Returns the instructions.Type
of val
.
tfp.experimental.auto_batching.type_inference.type_of_pattern(
val, backend, preferred_type=None
)
Args | |
---|---|
val
|
Pattern of backend-specific Tensor s or a Python or numpy constant.
|
backend
|
Object implementing required backend operations. |
preferred_type
|
instructions.Type to prefer, if t is a constant.
|
Returns | |
---|---|
vm_type
|
Pattern of instructions.TensorType describing t
|