Searched refs:outer_block_elems_a_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 388 Transpose<T, 1, transformation>(a, outer_block_elems_a_, b, in ExecuteTyped()
393 a, nodes.back().lda, outer_block_elems_a_, b, nodes.back().ldb, in ExecuteTyped()
399 Transpose<T, 2, transformation>(a, outer_block_elems_a_, b, in ExecuteTyped()
404 a, nodes.back().lda, outer_block_elems_a_, b, nodes.back().ldb, in ExecuteTyped()
411 Transpose<T, 4, transformation>(a, outer_block_elems_a_, b, in ExecuteTyped()
416 a, nodes.back().lda, outer_block_elems_a_, b, nodes.back().ldb, in ExecuteTyped()
422 Transpose<T, 8, transformation>(a, outer_block_elems_a_, b, in ExecuteTyped()
427 a, nodes.back().lda, outer_block_elems_a_, b, nodes.back().ldb, in ExecuteTyped()
433 Transpose<T, 16, transformation>(a, outer_block_elems_a_, b, in ExecuteTyped()
438 a, nodes.back().lda, outer_block_elems_a_, b, nodes.back().ldb, in ExecuteTyped()
[all …]
|
| H A D | transpose.h | 227 int outer_block_elems_a_ = 4; variable
|