Searched defs:ragged_tensor (Results 1 – 2 of 2) sorted by relevance
55 def _make_model(self, rank, width, ragged_tensor=False, flex=False): argument
265 const RaggedTensorVariant& ragged_tensor) { in ReturnRaggedTensor()