Searched defs:row_dims (Results 1 – 3 of 3) sorted by relevance
403 absl::Span<const int64_t> batch_dims, absl::Span<const int64_t> row_dims, in SetDotOperandLayout()425 absl::Span<const int64_t> batch_dims, absl::Span<const int64_t> row_dims, in SetOperandBatchRowsColsLayout()
70 absl::Span<const int64_t> row_dims, in GetBatchRowColumnShape()164 absl::Span<const int64_t> row_dims, absl::Span<const int64_t> col_dims) { in For()185 auto row_dims = in For() local
42 std::vector<int64_t> row_dims; member