Searched refs:total_input_size (Results 1 – 4 of 4) sorted by relevance
1054 int64 total_input_size = 0; in CalculateInputSize() local1057 total_input_size += input_size; in CalculateInputSize()1059 << " Total Input Size:" << total_input_size; in CalculateInputSize()1061 return total_input_size; in CalculateInputSize()1495 double total_input_size = 0; in PredictMaxPool() local1497 total_input_size = in PredictMaxPool()1504 total_input_size = in PredictMaxPool()1510 Costs costs = PredictOpCountBasedCost(ops, total_input_size, in PredictMaxPool()1545 double total_input_size = in PredictMaxPoolGrad() local1547 total_input_size += in PredictMaxPoolGrad()[all …]
177 int total_input_size = 1; in EvalPie() local179 total_input_size *= input->dims->data[i]; in EvalPie()183 const int batch_size = total_input_size / filter->dims->data[1]; in EvalPie()220 int total_input_size = 1; in EvalHybrid() local222 total_input_size *= input->dims->data[i]; in EvalHybrid()226 const int batch_size = total_input_size / filter->dims->data[1]; in EvalHybrid()238 if (tensor_utils::IsZeroVector(input->data.f, total_input_size)) { in EvalHybrid()
144 int total_input_size = 1; local146 total_input_size *= input.shape[i];148 input_size_ = total_input_size / batches_;280 int total_input_size = 1; local282 total_input_size *= input.shape[i];284 input_size_ = total_input_size / batches_;
686 total_input_size = 1689 total_input_size *= s694 x1 = [f * 1.0 for f in range(1, total_input_size + 1)]