Searched defs:to_tensor_spec (Results 1 – 2 of 2) sorted by relevance
45 def to_tensor_spec(meta: ScratchTensorMetadata) -> TensorSpec: function
277 def to_tensor_spec(input_spec, default_dtype=None): function