Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dsparse_matmul_op.cc1341 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in ComputeBlockSizes() local
1345 std::max<int>(1, left_dim0 / 64) * (left_dim1 / *KL) > est_num_cores) { in ComputeBlockSizes()
1434 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in Compute() local
1439 CHECK_EQ(left_dim0, 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()