Searched refs:rhs_array_reshaped (Results 1 – 1 of 1) sorted by relevance
948 llvm_ir::IrArray rhs_array_reshaped = in EmitBatchDotOperation() local967 dot_info.rhs_shape = DropFirstDim(rhs_array_reshaped.GetShape()); in EmitBatchDotOperation()983 SliceOutInnerArray(rhs_array_reshaped, /*batch_index=*/indvar, b); in EmitBatchDotOperation()