Searched defs:compute_shard (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_diag_op.cc | 368 auto compute_shard = [&output, &num_rows, &num_cols, &diag, in Compute() local
426 auto compute_shard = [&output, &input, &num_rows, &num_cols, in Compute() local
|
| D | matrix_set_diag_op.cc | 220 auto compute_shard = [&output, &diag, &upper_diag_index, &max_diag_len, in Compute() local
|
| D | matrix_band_part_op.cc | 150 auto compute_shard = [=, &input, &output](int64_t begin, int64_t end) { in operator ()() local
|