Searched refs:total_output_size (Results 1 – 2 of 2) sorted by relevance
1081 int64 total_output_size = 0; in CalculateOutputSize() local1093 total_output_size += output_size; in CalculateOutputSize()1095 << " Total Output Size:" << total_output_size; in CalculateOutputSize()1097 return total_output_size; in CalculateOutputSize()1507 const double total_output_size = in PredictMaxPool() local1511 total_output_size, op_info); in PredictMaxPool()1514 costs.max_memory = total_output_size; in PredictMaxPool()1550 const double total_output_size = in PredictMaxPoolGrad() local1554 total_output_size, op_info); in PredictMaxPoolGrad()1557 costs.max_memory = total_output_size; in PredictMaxPoolGrad()[all …]
1037 uint64 total_output_size = 0; in GenerateRunMetadata() local1048 total_output_size += tensor_size; in GenerateRunMetadata()1070 persistent_memory_size = total_output_size; in GenerateRunMetadata()