Searched refs:total_input_size (Results 1 – 10 of 10) sorted by relevance
31 int total_input_size = 1; in FullyConnectedOpModel() local33 total_input_size *= input.shape[i]; in FullyConnectedOpModel()35 input_size_ = total_input_size / batches_; in FullyConnectedOpModel()
357 int total_input_size = 1; local359 total_input_size *= lhs.shape[i];361 input_size_ = total_input_size / batches_;716 int total_input_size = 1; local718 total_input_size *= lhs.shape[i];720 input_size_ = total_input_size / batches_;
146 int total_input_size = 1; local148 total_input_size *= input.shape[i];150 input_size_ = total_input_size / batches_;302 int total_input_size = 1; local304 total_input_size *= input.shape[i];306 input_size_ = total_input_size / batches_;1192 int total_input_size = 1; in SparseFullyConnectedOpModel() local1194 total_input_size *= input.shape[i]; in SparseFullyConnectedOpModel()1196 input_size_ = total_input_size / batches_; in SparseFullyConnectedOpModel()
403 int total_input_size = 1; in EvalPie() local405 total_input_size *= input->dims->data[i]; in EvalPie()409 const int batch_size = total_input_size / filter->dims->data[1]; in EvalPie()
156 total_input_size = 1158 total_input_size *= s160 i * 1.0 / total_input_size for i in range(1, total_input_size + 1)
112 int total_input_size = ctx->InputShape(0).num_elements(); in CompileImpl() local114 int sample_size = total_input_size / total_scale_size; in CompileImpl()
71 int total_input_size = 1; local73 total_input_size *= input.shape[i];75 input_size_ = total_input_size / batches_;
782 total_input_size = 1785 total_input_size *= s790 x1 = [f * 1.0 for f in range(1, total_input_size + 1)]
1551 int64 total_input_size = 0; in CalculateInputSize() local1554 total_input_size += input_size; in CalculateInputSize()1556 << " Total Input Size:" << total_input_size; in CalculateInputSize()1558 return total_input_size; in CalculateInputSize()
1993 int total_input_size = bn_train_input_type_tensor.getNumElements(); in matchAndRewrite() local1995 int sample_size = total_input_size / total_scale_size; in matchAndRewrite()