Searched refs:lhs_array_reshaped (Results 1 – 1 of 1) sorted by relevance
946 llvm_ir::IrArray lhs_array_reshaped = in EmitBatchDotOperation() local953 int64 batch_count = lhs_array_reshaped.GetShape().dimensions(0); in EmitBatchDotOperation()966 dot_info.lhs_shape = DropFirstDim(lhs_array_reshaped.GetShape()); in EmitBatchDotOperation()981 SliceOutInnerArray(lhs_array_reshaped, /*batch_index=*/indvar, b); in EmitBatchDotOperation()