Searched refs:GetFirstDimensionSize (Results 1 – 2 of 2) sorted by relevance
325 Status GetFirstDimensionSize(OpKernelContext* context, INDEX_TYPE* result) { in GetFirstDimensionSize() function in tensorflow::__anonba91e2a20111::RaggedTensorToTensorBaseOp348 OP_REQUIRES_OK(context, GetFirstDimensionSize(context, &first_dimension)); in Compute()
85 int GetFirstDimensionSize(TfLiteContext* context, in GetFirstDimensionSize() function626 GetFirstDimensionSize(context, first_partition_input, attributes); in Eval()