HomeSort by relevance Sort by last modified time
    Searched refs:compute_shard (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
matrix_set_diag_op.cc 124 auto compute_shard = [&output, &diag](int64 begin, int64 end) { local
135 std::move(compute_shard));
matrix_band_part_op.cc 149 auto compute_shard = [=, &input, &output](int64 begin, int64 end) { local
188 thread_pool->ParallelFor(total_rows, row_cost, std::move(compute_shard));

Completed in 79 milliseconds