Searched refs:RecordInputBatchSize (Results 1 – 2 of 2) sorted by relevance
46 void RecordInputBatchSize(int32 batch_size, const string& model_name, in RecordInputBatchSize() function178 RecordInputBatchSize(tensors[0].shape().dim_size(0), GetModelName(context), in RegisterInput()
6066 func RecordInputBatchSize(value int64) RecordInputAttr { func