HomeSort by relevance Sort by last modified time
    Searched defs:lhs_rank (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
dot_decomposer.cc 173 const int64 lhs_rank = lhs_shape.rank(); local
175 lhs_rank - num_batch_dims - num_contracting_dims;
183 for (int64 i = 0; i < lhs_rank; ++i) {
197 lhs_transpose.reserve(lhs_rank);
indexed_array_analysis.cc 1045 int64 lhs_rank = lhs->shape().rank(); local
    [all...]

Completed in 257 milliseconds