Searched refs:GetRowPartitionTensor (Results 1 – 2 of 2) sorted by relevance
78 RowPartitionTensor GetRowPartitionTensor(OpKernelContext* c, int dimension) { in GetRowPartitionTensor() function in tensorflow::__anonba91e2a20111::RaggedTensorToTensorBaseOp89 GetRowPartitionTensor(c, dimension - 1); in GetMaxWidth()305 GetRowPartitionTensor(context, dimension); in CalculateOutputIndex()
142 const TfLiteTensor* GetRowPartitionTensor( in GetRowPartitionTensor() function215 const TfLiteTensor* tensor = GetRowPartitionTensor( in GetMaxWidth()429 GetRowPartitionTensor(conversion_attributes, context, node, dimension); in CalculateOutputIndex()