Searched refs:target_array_reshaped (Results 1 – 1 of 1) sorted by relevance
1343 llvm_ir::IrArray target_array_reshaped = in PotentiallyImplementedAsEigenMatmul() local1353 dot_info.result_shape = DropFirstDim(target_array_reshaped.GetShape()); in PotentiallyImplementedAsEigenMatmul()1415 llvm_ir::IrArray target_array_reshaped = in EmitBatchDotOperation() local1435 DropFirstDim(target_array_reshaped.GetShape()); in EmitBatchDotOperation()1452 target_array_reshaped, /*batch_index=*/indvar, b); in EmitBatchDotOperation()