Searched refs:outer_block_elems_b_ (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | transpose.cc | 389 outer_block_elems_b_, nodes.data(), in ExecuteTyped()
394 outer_block_elems_b_, scratch.get()); in ExecuteTyped()
400 outer_block_elems_b_, nodes.data(), in ExecuteTyped()
405 outer_block_elems_b_, scratch.get()); in ExecuteTyped()
412 outer_block_elems_b_, nodes.data(), in ExecuteTyped()
417 outer_block_elems_b_, scratch.get()); in ExecuteTyped()
423 outer_block_elems_b_, nodes.data(), in ExecuteTyped()
428 outer_block_elems_b_, scratch.get()); in ExecuteTyped()
434 outer_block_elems_b_, nodes.data(), in ExecuteTyped()
439 outer_block_elems_b_, scratch.get()); in ExecuteTyped()
[all …]
|
| H A D | transpose.h | 228 int outer_block_elems_b_ = 4; variable
|