Searched refs:lhs_transposed_shape (Results 1 – 2 of 2) sorted by relevance
345 TensorShape lhs_transposed_shape( in TransposeLhsForGtsvBatched() local351 lhs_transposed_shape, &lhs_transposed)); in TransposeLhsForGtsvBatched()
471 llvm::SmallVector<int64_t, 4> lhs_transposed_shape = Concat<int64_t>( in ConvertDot() local477 RankedTensorType::get(lhs_transposed_shape, lhs_type.getElementType()), in ConvertDot()