Searched refs:SetOutputBatchDimensions (Results 1 – 2 of 2) sorted by relevance
47 void SetOutputBatchDimensions(TfLiteIntArray* output_batch_dimensions) { in SetOutputBatchDimensions() function
63 builder->SetOutputBatchDimensions(output_batch_dimensions); in CreateBatchSeqBuilder()