Searched refs:num_total_output_bytes (Results 1 – 2 of 2) sorted by relevance
61 int64 num_total_output_bytes() const { in num_total_output_bytes() function66 return num_total_output_bytes() + internal_write_bytes; in num_total_write_bytes()
1637 node_costs->max_memory = node_costs->num_total_output_bytes(); in PredictDefaultNodeCosts()2363 node_costs->max_memory = node_costs->num_total_output_bytes(); in PredictAvgPoolGrad()2408 node_costs->max_memory = node_costs->num_total_output_bytes(); in PredictFusedBatchNorm()2445 node_costs->max_memory = node_costs->num_total_output_bytes(); in PredictFusedBatchNormGrad()