Searched defs:RaggedTensorSpec (Results 1 – 3 of 3) sorted by relevance
1572 ragged_tensor.RaggedTensorSpec, StructuredTensor.Spec,
2309 class RaggedTensorSpec(type_spec.BatchableTypeSpec): class
1309 spec: Union["DynamicRaggedShape.Spec", ragged_tensor.RaggedTensorSpec,