Searched refs:size_nhwc (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | op_level_cost_estimator.cc | 2390 const int64 size_nhwc = in PredictFusedBatchNorm() local
2395 node_costs->num_input_bytes_accessed = {size_nhwc, size_c, size_c}; in PredictFusedBatchNorm()
2396 node_costs->num_output_bytes_accessed = {size_nhwc, size_c, size_c, size_c, in PredictFusedBatchNorm()
2402 node_costs->internal_read_bytes = size_nhwc; in PredictFusedBatchNorm()
2404 node_costs->num_input_bytes_accessed = {size_nhwc, size_c, size_c, size_c, in PredictFusedBatchNorm()
2406 node_costs->num_output_bytes_accessed = {size_nhwc}; in PredictFusedBatchNorm()
2435 const int64 size_nhwc = in PredictFusedBatchNormGrad() local
2441 node_costs->num_input_bytes_accessed = {size_nhwc, size_nhwc, size_c, size_c}; in PredictFusedBatchNormGrad()
2442 node_costs->num_output_bytes_accessed = {size_nhwc, size_c, size_c}; in PredictFusedBatchNormGrad()
2444 node_costs->internal_read_bytes = size_nhwc; in PredictFusedBatchNormGrad()
|