Searched refs:keras_tensor_subclass (Results 1 – 1 of 1) sorted by relevance
591 def register_keras_tensor_specialization(cls, keras_tensor_subclass): argument594 keras_tensor_classes.insert(-1, (cls, keras_tensor_subclass))