Searched refs:transpose_left (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_matmul_op.cc | 796 const ConstMatrixMapR& right, bool transpose_left,
852 bool transpose_left, int num_threads,
944 const ConstMatrixMapR& right, bool transpose_left,
1319 bool transpose_left, int num_threads, int* KR, int* NR, int* KL, int* JB, in ComputeBlockSizes() argument
1340 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in ComputeBlockSizes()
1341 const int left_dim1 = transpose_left ? left.dimension(0) : left.dimension(1); in ComputeBlockSizes()
1429 bool transpose_left, const DeviceBase::CpuWorkerThreads* thread_pool, in Compute() argument
1435 transpose_left, thread_pool, transpose_output, output); in Compute()
1439 const int left_dim0 = transpose_left ? left.dimension(1) : left.dimension(0); in Compute()
1440 const int left_dim1 = transpose_left ? left.dimension(0) : left.dimension(1); in Compute()
[all …]
|