Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1441 const int right_dim0 = right.dimension(0); in Compute() local
1443 CHECK_EQ(left_dim1, right_dim0); in Compute()
1461 cache->take_cache_entry(left_dim0, right_dim0, right_dim1, num_threads); in Compute()
1547 const int right_dim0 = right.dimension(0); in Compute() local
1553 std::min(KR, right_dim0) * (std::min(NR, right_dim1) + N - 1) / N; in Compute()
1560 const int num_k_blocks = (right_dim0 + KR - 1) / KR; in Compute()
1569 std::min(KR, static_cast<int>(right_dim0 - KR * kb)); in Compute()