Searched refs:ldb_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | transpose.cc | 759 : ldb_[pos_stride1a_in_b]; in BuildPlanNodes()
785 node.ldb = b_tiling_[b_dim] > 1 ? ldb_tile_[b_dim] : ldb_[b_dim]; in BuildPlanNodes()
838 node.ldb = ldb_[b_dim] * tile_size / b_tiling_[b_dim]; in BuildPlanNodes()
1065 ComputeStrides(elem_size_in_bytes_, b_dims_, b_tiling_, ldb_, ldb_tile_); in Initialize()
1158 (l.tile_interior && b_is_tiled_) ? ldb_tile_[b_dim] : ldb_[b_dim]; in Initialize()
1320 absl::StrJoin(lda_tile_, ","), absl::StrJoin(ldb_, ","), in ToString()
|
| D | transpose.h | 194 absl::InlinedVector<int64_t, 4> ldb_; variable
|