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