Searched refs:size_nhwc (Results 1 – 1 of 1) sorted by relevance
1669 const double size_nhwc = in PredictFusedBatchNorm() local1677 total_input_size = size_nhwc + size_c * 2; in PredictFusedBatchNorm()1678 total_output_size = size_nhwc + size_c * 4; in PredictFusedBatchNorm()1679 total_internal_read_size = size_nhwc; in PredictFusedBatchNorm()1681 total_input_size = size_nhwc + size_c * 4; in PredictFusedBatchNorm()1682 total_output_size = size_nhwc; in PredictFusedBatchNorm()1711 const double size_nhwc = in PredictFusedBatchNormGrad() local1715 double total_input_size = size_nhwc * 2 + size_c * 2; in PredictFusedBatchNormGrad()1716 double total_internal_read_size = size_nhwc; in PredictFusedBatchNormGrad()1717 double total_output_size = size_nhwc * 1 + size_c * 2; in PredictFusedBatchNormGrad()