Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dquantized_matmul_op.cc115 int a_dim_remaining = 1 - dim_pair[0].first; in Compute() local
118 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()
128 const size_t m = a.dim_size(a_dim_remaining); in Compute()
Dmkl_matmul_op.cc70 int a_dim_remaining = 1 - dim_pair[0].first; in Compute() local
73 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()
Dmatmul_op.cc473 int a_dim_remaining = 1 - dim_pair[0].first; in Compute() local
476 {a.dim_size(a_dim_remaining), b.dim_size(b_dim_remaining)}); in Compute()