Home
last modified time | relevance | path

Searched refs:right_dim1 (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1437 const int right_dim1 = right.dimension(1); in Compute() local
1441 CHECK_EQ(right_dim1, in Compute()
1444 if (left_dim0 < 32 || left_dim1 < 32 || right_dim1 < 32) { in Compute()
1458 cache->take_cache_entry(left_dim0, right_dim0, right_dim1, num_threads); in Compute()
1545 const int right_dim1 = right.dimension(1); in Compute() local
1550 std::min(KR, right_dim0) * ((std::min(NR, right_dim1) + N - 1) / N); in Compute()
1558 const int num_n_blocks = (right_dim1 + NR - 1) / NR; in Compute()
1563 std::min(NR, static_cast<int>(right_dim1 - NR * nb)); in Compute()