Searched defs:cpu_budget (Results 1 – 3 of 3) sorted by relevance
52 Dataset(OpKernelContext* ctx, const DatasetBase* input, int64 cpu_budget) in Dataset()
410 void Model::Optimize(int64 cpu_budget) { in Optimize()
3286 def __init__(self, input_dataset, cpu_budget): argument