Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op.cc116 int b_dim_remaining = 1 - dim_pair[0].second; in Compute() local
118 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()
129 const size_t n = b.dim_size(b_dim_remaining); in Compute()
Dmkl_matmul_op.cc71 int b_dim_remaining = 1 - dim_pair[0].second; in Compute() local
73 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()
Dmatmul_op.cc474 int b_dim_remaining = 1 - dim_pair[0].second; in Compute() local
476 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()