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.cc1442 const int right_dim1 = right.dimension(1); in Compute() local
1446 CHECK_EQ(right_dim1, in Compute()
1448 if (left_dim0 < 32 || left_dim1 < 32 || right_dim1 < 32) { in Compute()
1461 cache->take_cache_entry(left_dim0, right_dim0, right_dim1, num_threads); in Compute()
1548 const int right_dim1 = right.dimension(1); in Compute() local
1553 std::min(KR, right_dim0) * (std::min(NR, right_dim1) + N - 1) / N; in Compute()
1561 const int num_n_blocks = (right_dim1 + NR - 1) / NR; in Compute()
1566 std::min(NR, static_cast<int>(right_dim1 - NR * nb)); in Compute()