Searched refs:num_output_dims (Results 1 – 3 of 3) sorted by relevance
92 const int num_output_dims = output->dims->size; in FillDiagHelper() local94 for (int i = 0; i < num_output_dims - 2; ++i) { in FillDiagHelper()98 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper()99 const int col_size = output->dims->data[num_output_dims - 1]; in FillDiagHelper()
96 const int num_output_dims = output->dims->size; in FillDiagHelper() local98 for (int i = 0; i < num_output_dims - 2; ++i) { in FillDiagHelper()102 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper()103 const int col_size = output->dims->data[num_output_dims - 1]; in FillDiagHelper()
48 const int num_output_dims = output->dims(); in operator() local52 for (int i = 0; i < num_dims - num_output_dims; ++i) in operator()54 for (int i = num_dims - num_output_dims; i < num_dims; ++i) in operator()217 const int num_output_dims = output->dims(); in operator() local222 for (int i = axis_begin_dim; i < axis_begin_dim + num_output_dims; ++i) in operator()224 for (int i = axis_begin_dim + num_output_dims; i < num_dims; ++i) in operator()