Searched defs:lda_tile (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | transpose_test.cc | 48 absl::InlinedVector<int64_t, 4> lda_tile = {1, 1, 1, 1, 1, 1, 1}; in TEST() local
72 absl::InlinedVector<int64_t, 4> lda_tile = {1, 1, 1, 1, 1, 1, 1}; in TEST() local
|
| D | transpose.cc | 537 absl::InlinedVector<int64_t, 4>& lda_tile, in RemoveTrivialDimensions()
593 absl::InlinedVector<int64_t, 4>& lda_tile, in CoalesceDimensions()
|