Searched refs:max_memory (Results 1 – 6 of 6) sorted by relevance
22 CHECK_NE(left.max_memory, kMemoryUnknown); in CombineCosts()48 if (right.max_memory != kMemoryUnknown) { in CombineCosts()49 result.max_memory += right.max_memory; in CombineCosts()74 if (result.max_memory != kMemoryUnknown) { in MultiplyCosts()75 result.max_memory *= multiplier; in MultiplyCosts()
32 c.max_memory = 1; in TEST()47 EXPECT_EQ(sum.max_memory, 2); in TEST()63 c.max_memory = 1; in TEST()78 EXPECT_EQ(product.max_memory, 10); in TEST()
141 int64 max_memory; // Maximum main memory requirement in bytes over all ops. member184 max_memory = kMemoryUnknown; in Costs()197 costs.max_memory = kZeroMemory; in ZeroCosts()
506 costs.max_memory = output_size; in PredictOpCountBasedCost()1272 costs.max_memory = output_size; in PredictSparseTensorDenseMatMul()1287 result.max_memory = CalculateOutputSize(op_info, &result.inaccurate); in PredictIdentity()1322 costs.max_memory = CalculateOutputSize(op_info, &costs.inaccurate); in PredictMetadata()1369 costs.max_memory = output_size; in PredictGatherOrSlice()1514 costs.max_memory = total_output_size; in PredictMaxPool()1557 costs.max_memory = total_output_size; in PredictMaxPoolGrad()1590 costs.max_memory = total_output_size; in PredictAvgPool()1643 costs.max_memory = total_output_size; in PredictAvgPoolGrad()1690 costs.max_memory = total_output_size; in PredictFusedBatchNorm()[all …]
868 VLOG(1) << "Expected max memory: " << graph_costs_.max_memory; in Summary()
434 * counted against the JPEG max_memory setting. In some systems the above