Searched refs:lda_ (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | transpose.cc | 756 : lda_[pos_stride1b_in_a]; in BuildPlanNodes()
784 node.lda = a_tiling_[a_dim] > 1 ? lda_tile_[a_dim] : lda_[a_dim]; in BuildPlanNodes()
837 node.lda = lda_[a_dim] * tile_size / a_tiling_[a_dim]; in BuildPlanNodes()
970 plan->lda_.reserve(ndim); in Create()
974 plan->lda_.push_back(input_strides_in_bytes.at(dim_order[i])); in Create()
988 plan->lda_, plan->lda_tile_); in Create()
1015 plan->lda_.back() != sizeof(float)) { in Create()
1032 RemoveTrivialDimensions(a_dims_, permutation_, lda_, lda_tile_, a_tiling_, in Initialize()
1034 CoalesceDimensions(a_dims_, permutation_, lda_, lda_tile_, a_tiling_, in Initialize()
1044 lda_.empty() in Initialize()
[all …]
|
| D | transpose.h | 192 absl::InlinedVector<int64_t, 4> lda_; variable
|