Searched refs:transpose_left (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 795 const ConstMatrixMapR& right, bool transpose_left,
851 bool transpose_left, int num_threads,
943 const ConstMatrixMapR& right, bool transpose_left,
1320 bool transpose_left, int num_threads, int* KR, int* NR, int* KL, int* JB, in ComputeBlockSizes() argument
1341 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in ComputeBlockSizes()
1342 const int left_dim1 = transpose_left ? left.dimension(0) : left.dimension(1); in ComputeBlockSizes()
1431 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute() argument
1434 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in Compute()
1435 const int left_dim1 = transpose_left ? left.dimension(0) : left.dimension(1); in Compute()
1448 transpose_left, thread_pool, transpose_output, output); in Compute()
[all …]
|