Searched refs:N_full (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_fill_empty_rows_op.cc | 139 const Tindex N_full = csr_offset[dense_rows - 1]; in operator ()() local
140 TensorShape output_indices_shape({N_full, rank}); in operator ()()
147 kOutputValuesOutput, TensorShape({N_full}), &output_values_t)); in operator ()()
250 const Tindex N_full = grad_values.dimension(0); in operator ()() local
257 context->allocate_temp(DT_BOOL, TensorShape({N_full}), &visited_t)); in operator ()()
266 if (reverse_index < 0 || reverse_index >= N_full) { in operator ()()
268 "Elements in reverse index must be in [0, ", N_full, ") but got ", in operator ()()
274 for (int j = 0; j < N_full; ++j) { in operator ()()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseFillEmptyRowsGrad.pbtxt | 30 shaped `[N_full]`, where `N_full >= N` and copies data into either
35 d_default_value = sum_{k : 0 .. N_full - 1} (
|